diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 95182e0179..83be0dddd6 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -222,7 +222,9 @@ class _RaiseUninitialized(logging.Handler): super().__init__() def emit(self, record): record_lower = str(record).lower() - if "some weights of" in record_lower and "score.weight" not in record_lower: + if ("some weights of" in record_lower) and \ + ("score.weight" not in record_lower) and \ + ("classifier.weight" not in record_lower): raise Exception( f"Unsloth: Critical error since some weights are not initialized.\n"\ f"Please try updating Unsloth, transformers and timm via:\n"\