Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-22 06:08:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

remove unused class CrossAttention in SAM

Browse source
This commit is contained in:
Pierre Chapuis 2024-01-29 16:34:35 +01:00 committed by Cédric Deltheil
parent a1ad317b00
commit 86867e9318
1 changed files with 0 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
src/refiners/foundationals/segment_anything/transformer.py
View file

@ -3,29 +3,6 @@ from torch import device as Device, dtype as DType
import refiners.fluxion.layers as fl import refiners.fluxion.layers as fl
class CrossAttention(fl.Attention):
def __init__(
self,
embedding_dim: int,
cross_embedding_dim: int | None = None,
num_heads: int = 1,
inner_dim: int | None = None,
device: Device | str | None = None,
dtype: DType | None = None,
) -> None:
super().__init__(
embedding_dim=embedding_dim,
key_embedding_dim=cross_embedding_dim,
num_heads=num_heads,
inner_dim=inner_dim,
is_optimized=False,
device=device,
dtype=dtype,
)
self.cross_embedding_dim = cross_embedding_dim or embedding_dim
self.insert(index=0, module=fl.Parallel(fl.GetArg(index=0), fl.GetArg(index=1), fl.GetArg(index=1)))
class FeedForward(fl.Residual): class FeedForward(fl.Residual):
def __init__( def __init__(
self, embedding_dim: int, feed_forward_dim: int, device: Device | str | None = None, dtype: DType | None = None self, embedding_dim: int, feed_forward_dim: int, device: Device | str | None = None, dtype: DType | None = None