Update app.py
Browse files
app.py
CHANGED
|
@@ -417,7 +417,7 @@ def run_tuning_task():
|
|
| 417 |
os.makedirs(local_dir, exist_ok=True)
|
| 418 |
|
| 419 |
# Save model using Hugging Face format
|
| 420 |
-
merged_model.save_pretrained(local_dir, safe_serialization=
|
| 421 |
tokenizer.save_pretrained(local_dir)
|
| 422 |
|
| 423 |
# Push to Hub
|
|
|
|
| 417 |
os.makedirs(local_dir, exist_ok=True)
|
| 418 |
|
| 419 |
# Save model using Hugging Face format
|
| 420 |
+
merged_model.save_pretrained(local_dir, safe_serialization=False)
|
| 421 |
tokenizer.save_pretrained(local_dir)
|
| 422 |
|
| 423 |
# Push to Hub
|