Update app_flash.py
Browse files- 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.
|
| 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.")
|