Hanna commited on
Commit
2dcf48b
·
unverified ·
1 Parent(s): fc946a1

Fix in-place addition of patch_embds (#85)

Browse files
Files changed (1) hide show
  1. bytelatent/model/local_models.py +1 -2
bytelatent/model/local_models.py CHANGED
@@ -308,8 +308,7 @@ class LocalEncoder(LocalModelBase):
308
  kv=h,
309
  mask=cross_mask,
310
  )
311
- patch_embeds += patch_embeds_cross
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):