Spaces:
Sleeping
Sleeping
Update train.py
Browse files
train.py
CHANGED
|
@@ -51,5 +51,3 @@ trainer.train()
|
|
| 51 |
# Save the model
|
| 52 |
model.save_pretrained("./tuned_model")
|
| 53 |
tokenizer.save_pretrained("./tuned_model")
|
| 54 |
-
|
| 55 |
-
|
|
|
|
| 51 |
# Save the model
|
| 52 |
model.save_pretrained("./tuned_model")
|
| 53 |
tokenizer.save_pretrained("./tuned_model")
|
|
|
|
|
|