Spaces:
Paused
Paused
Update train.py
Browse files
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
|
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")
|
|
|
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")
|