saving

unsloth/models/mistral.py

@@ -375,6 +375,9 @@ class FastMistralModel(FastLlamaModel):
         
         # Log Unsloth version for future fastpaths for inference
         model.config.update({"unsloth_version" : __version__})
+
+        # Add save modules
+        patch_saving_functions(model)
         
         return model, tokenizer
     pass

unsloth/save.py

@@ -278,7 +278,6 @@ def unsloth_save_model(
         not hasattr(internal_model.model, "layers")
     ):
         # Do general saving
-        print(type(model))
         # Edit save_pretrained_settings
         # [TODO] _create_repo has errors due to **kwargs getting accepted
         for deletion in \