rahul7star commited on
Commit
74442ab
·
verified ·
1 Parent(s): 8191e5c

Update app_flash.py

Browse files
Files changed (1) hide show
  1. 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