mirror of
https://github.com/facebookresearch/blt.git
synced 2025-04-07 10:19:08 +00:00
Fix in-place addition of patch_embds (#85)
This commit is contained in:
parent
fc946a1918
commit
2dcf48bdd9
1 changed files with 1 additions and 2 deletions
|
@ -308,8 +308,7 @@ class LocalEncoder(LocalModelBase):
|
||||||
kv=h,
|
kv=h,
|
||||||
mask=cross_mask,
|
mask=cross_mask,
|
||||||
)
|
)
|
||||||
patch_embeds += patch_embeds_cross
|
return patch_embeds + patch_embeds_cross
|
||||||
return patch_embeds
|
|
||||||
|
|
||||||
|
|
||||||
class LocalDecoder(LocalModelBase):
|
class LocalDecoder(LocalModelBase):
|
||||||
|
|
Loading…
Add table
Reference in a new issue