From 2dcf48bdd96bf7d3d868e5d799534eda8774b7fa Mon Sep 17 00:00:00 2001
From: Hanna <53936399+geranium12@users.noreply.github.com>
Date: Fri, 21 Mar 2025 00:46:32 +0100
Subject: [PATCH] Fix in-place addition of patch_embds (#85)
---
bytelatent/model/local_models.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/bytelatent/model/local_models.py b/bytelatent/model/local_models.py
index 09a5a19..7083ac4 100644
--- a/bytelatent/model/local_models.py
+++ b/bytelatent/model/local_models.py
@@ -308,8 +308,7 @@ class LocalEncoder(LocalModelBase):
kv=h,
mask=cross_mask,
)
- patch_embeds += patch_embeds_cross
- return patch_embeds
+ return patch_embeds + patch_embeds_cross
class LocalDecoder(LocalModelBase):