diff --git a/src/axolotl/utils/models.py b/src/axolotl/utils/models.py index 523fd76fe..4a665c111 100644 --- a/src/axolotl/utils/models.py +++ b/src/axolotl/utils/models.py @@ -1057,7 +1057,7 @@ class ModelLoader: ) if ( hasattr(self.model, "get_input_embeddings") - and self.model.get_input_embeddings().num_embeddings < embeddings_len + and self.model.get_input_embeddings().num_embeddings != embeddings_len ): resize_kwargs = {} if self.cfg.mean_resizing_embeddings is not None: