rahul7star commited on
Commit
fad9b48
·
verified ·
1 Parent(s): 9b6142b

Update app_flash.py

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