Update app_flash.py
Browse files- app_flash.py +1 -1
app_flash.py
CHANGED
|
@@ -76,7 +76,7 @@ def push_flashpack_to_hub_local(model, hf_repo: str, size_limit_mb=100):
|
|
| 76 |
|
| 77 |
# 1️⃣ Save model locally to temp directory
|
| 78 |
temp_dir = tempfile.mkdtemp()
|
| 79 |
-
model.save_flashpack(temp_dir)
|
| 80 |
logs.append(f"📦 Model saved locally to {temp_dir}")
|
| 81 |
|
| 82 |
# 2️⃣ List saved files
|
|
|
|
| 76 |
|
| 77 |
# 1️⃣ Save model locally to temp directory
|
| 78 |
temp_dir = tempfile.mkdtemp()
|
| 79 |
+
model.save_flashpack(temp_dir,target_dtype=torch.float32)
|
| 80 |
logs.append(f"📦 Model saved locally to {temp_dir}")
|
| 81 |
|
| 82 |
# 2️⃣ List saved files
|