nroggendorff commited on
Commit
6b9cbb9
·
verified ·
1 Parent(s): 8fd094c

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +2 -2
train.py CHANGED
@@ -133,8 +133,8 @@ def train_model(model, tokenizer, dataset, push):
133
  if push:
134
  repo_id = OUTPUT_REPO
135
  msg = str(train.training_loss)
136
- trained_model.push_to_hub(repo_id, commit_message=msg, force=True)
137
- trained_tokenizer.push_to_hub(repo_id, commit_message=msg, force=True)
138
  else:
139
  trained_model.save_pretrained("model")
140
  trained_tokenizer.save_pretrained("tokenizer")
 
133
  if push:
134
  repo_id = OUTPUT_REPO
135
  msg = str(train.training_loss)
136
+ trained_model.push_to_hub(repo_id, commit_message=msg)
137
+ trained_tokenizer.push_to_hub(repo_id, commit_message=msg)
138
  else:
139
  trained_model.save_pretrained("model")
140
  trained_tokenizer.save_pretrained("tokenizer")