Keeby-smilyai commited on
Commit
4add9c8
·
verified ·
1 Parent(s): be1329a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=True)
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