Spaces:
Running
on
Zero
Running
on
Zero
Hanna
commited on
Fix in-place addition of patch_embds (#85)
Browse files
bytelatent/model/local_models.py
CHANGED
@@ -308,8 +308,7 @@ class LocalEncoder(LocalModelBase):
|
|
308 |
kv=h,
|
309 |
mask=cross_mask,
|
310 |
)
|
311 |
-
patch_embeds
|
312 |
-
return patch_embeds
|
313 |
|
314 |
|
315 |
class LocalDecoder(LocalModelBase):
|
|
|
308 |
kv=h,
|
309 |
mask=cross_mask,
|
310 |
)
|
311 |
+
return patch_embeds + patch_embeds_cross
|
|
|
312 |
|
313 |
|
314 |
class LocalDecoder(LocalModelBase):
|