rahul7star commited on
Commit
0b52d90
·
verified ·
1 Parent(s): 920ad39

Update app_flash.py

Browse files
Files changed (1) hide show
  1. app_flash.py +1 -1
app_flash.py CHANGED
@@ -94,7 +94,7 @@ def push_flashpack_model_to_hf(model, hf_repo: str):
94
  logs.append(f"🌐 Hugging Face repo cloned to: {tmp_dir}")
95
 
96
  # Save model inside repo
97
- pack_path = os.path.join(tmp_dir, "model.pack")
98
  logs.append(f"💾 Saving model to: {pack_path}")
99
  model.save_flashpack(pack_path, target_dtype=torch.float32)
100
  logs.append("✅ Model saved successfully.")
 
94
  logs.append(f"🌐 Hugging Face repo cloned to: {tmp_dir}")
95
 
96
  # Save model inside repo
97
+ pack_path = os.path.join(tmp_dir, "model.flashpack")
98
  logs.append(f"💾 Saving model to: {pack_path}")
99
  model.save_flashpack(pack_path, target_dtype=torch.float32)
100
  logs.append("✅ Model saved successfully.")