don't resize embeddings if it's already large enough (#577)

* don't resize embeddings if it's already large enough * make sure to tie weights, even if we aren't resizing
axolotl-ai-cloud · Sep 15, 2023 · 6ecd91e · 6ecd91e
1 parent 1ebc6fc
commit 6ecd91e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/axolotl/utils/models.py b/src/axolotl/utils/models.py
@@ -302,7 +302,10 @@ def load_model(
         if cfg.resize_token_embeddings_to_32x
         else len(tokenizer)
     )
-    model.resize_token_embeddings(embeddings_len)
+    if model.get_input_embeddings().num_embeddings < embeddings_len:
+        model.resize_token_embeddings(embeddings_len)
+    else:
+        model.tie_weights()
 
     if (
         hasattr(model.config, "max_position_embeddings")