vikarti.anatra/blt
1
0
Fork 0
mirror of https://github.com/facebookresearch/blt.git synced 2025-09-01 10:09:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix in-place addition of patch_embds

Browse source
This commit is contained in:
Hanna Herasimchyk 2025-03-18 16:34:05 +01:00
parent fc946a1918
commit b460ef3cd3
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
bytelatent/model/local_models.py
View file

@ -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):