mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-15 01:28:14 +00:00
fix typos
This commit is contained in:
parent
d90bb25151
commit
831bc7f2db
|
@ -80,7 +80,7 @@ class SegmentAnything(fl.Chain):
|
||||||
|
|
||||||
@no_grad()
|
@no_grad()
|
||||||
def compute_image_embedding(self, image: Image.Image) -> ImageEmbedding:
|
def compute_image_embedding(self, image: Image.Image) -> ImageEmbedding:
|
||||||
"""Compute the emmbedding of an image.
|
"""Compute the embedding of an image.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
image: The image to compute the embedding of.
|
image: The image to compute the embedding of.
|
||||||
|
|
|
@ -205,7 +205,7 @@ class WindowSDPA(fl.Module):
|
||||||
|
|
||||||
class WindowAttention(fl.Chain):
|
class WindowAttention(fl.Chain):
|
||||||
"""
|
"""
|
||||||
Window-based Multi-head Self-Attenion (W-MSA), optionally shifted (SW-MSA).
|
Window-based Multi-head Self-Attention (W-MSA), optionally shifted (SW-MSA).
|
||||||
|
|
||||||
It has a trainable relative position bias (RelativePositionBias).
|
It has a trainable relative position bias (RelativePositionBias).
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue