Change resize_token_embeddings to make it return same Class that is p…

…assed to it
huggingface · Jul 16, 2024 · d9e1b5e · d9e1b5e
1 parent 6fbea6d
commit d9e1b5e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py
@@ -2096,7 +2096,8 @@ def _get_resized_embeddings(
         else:
             new_embeddings.weight.data[:n, :] = old_embeddings.weight.data[:n, :]
 
-        return new_embeddings
+        old_embeddings.weight.data = new_embeddings.weight.data
+        return old_embeddings
 
     def _get_resized_lm_head(
         self, old_lm_head: nn.Linear, new_num_tokens: Optional[int] = None, transposed: Optional[bool] = False