Fix

huggingface · Sep 16, 2024 · 41a4d8a · 41a4d8a
1 parent 3e677c9
commit 41a4d8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/mllama/convert_mllama_weights_to_hf.py b/src/transformers/models/mllama/convert_mllama_weights_to_hf.py
@@ -400,7 +400,7 @@ def write_model(
     del model.config._name_or_path
 
     print("Saving the model.")
-    # model.save_pretrained(model_path, safe_serialization=safe_serialization)
+    model.save_pretrained(model_path, safe_serialization=safe_serialization)
     del state_dict, model
 
     # Safety check: reload the converted model