Update app_flash.py
Browse files- app_flash.py +2 -1
app_flash.py
CHANGED
|
@@ -126,7 +126,8 @@ def train_and_push_flashpack(
|
|
| 126 |
|
| 127 |
if push_to_hub:
|
| 128 |
print(f"📤 Pushing model to Hugging Face repo: {hf_repo} ...")
|
| 129 |
-
model.save_flashpack(hf_repo, target_dtype=torch.float32
|
|
|
|
| 130 |
print(f"✅ Model pushed to HF repo: {hf_repo}")
|
| 131 |
|
| 132 |
return model, dataset, embed_model, tokenizer, long_embeddings
|
|
|
|
| 126 |
|
| 127 |
if push_to_hub:
|
| 128 |
print(f"📤 Pushing model to Hugging Face repo: {hf_repo} ...")
|
| 129 |
+
model.save_flashpack(hf_repo, target_dtype=torch.float32) # save locally
|
| 130 |
+
model.push_to_hub(hf_repo) # then push to HF repo
|
| 131 |
print(f"✅ Model pushed to HF repo: {hf_repo}")
|
| 132 |
|
| 133 |
return model, dataset, embed_model, tokenizer, long_embeddings
|