Fix: Ensure tie_weights method is called only if it exists in the model class

This commit is contained in:
mhenrhcsen
2025-07-16 21:03:49 +02:00
parent 36cbe13d18
commit 84ad69afad

View File

@@ -235,7 +235,8 @@ class ModelLoader:
resize_kwargs["mean_resizing"] = self.cfg.mean_resizing_embeddings
self.model.resize_token_embeddings(embeddings_len, **resize_kwargs)
else:
self.model.tie_weights()
if hasattr(self.model, "tie_weights"):
self.model.tie_weights()
def _adjust_model_config(self):
if (