typo

eole-nlp · Jun 14, 2024 · faf83ee · faf83ee
1 parent 4954c12
commit faf83ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eole/models/model.py b/eole/models/model.py
@@ -562,7 +562,7 @@ def _load_param(self, name, module, param_name, param, buf_list, ckpt_t, offset)
             col_slice_start = 0
             col_slice_end = param.data.size(0)
         if param.data.dim() == 2:
-            if name.split(".")[-1] in ["final_linear", "up_proj"]:
+            if name.split(".")[-1] in ["final_linear", "down_proj"]:
                 row_slice_start = param.data.size(1) * offset
                 row_slice_end = param.data.size(1) * (offset + 1)
             else: