Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -105,7 +105,7 @@ print ("lm datasets")
|
|
| 105 |
batch_size = 2
|
| 106 |
|
| 107 |
training_args = TrainingArguments(
|
| 108 |
-
output_dir="alexkueck/test-tis",
|
| 109 |
overwrite_output_dir = 'True',
|
| 110 |
per_device_train_batch_size=batch_size, #batch_size = 2 for full training
|
| 111 |
per_device_eval_batch_size=batch_size,
|
|
@@ -172,14 +172,14 @@ print("Output:\n" )
|
|
| 172 |
|
| 173 |
|
| 174 |
print("Save to Space")
|
| 175 |
-
trainer.save_model("alexkueck/test-tis")
|
| 176 |
print("done")
|
| 177 |
|
| 178 |
#####################################
|
| 179 |
#Push to Hub
|
| 180 |
print("push to hub")
|
| 181 |
login(token=os.environ["HF_ACCESS_TOKEN"])
|
| 182 |
-
trainer.push_to_hub("alexkueck/test-tis")
|
| 183 |
print("done")
|
| 184 |
|
| 185 |
|
|
|
|
| 105 |
batch_size = 2
|
| 106 |
|
| 107 |
training_args = TrainingArguments(
|
| 108 |
+
output_dir="alexkueck/test-tis-1",
|
| 109 |
overwrite_output_dir = 'True',
|
| 110 |
per_device_train_batch_size=batch_size, #batch_size = 2 for full training
|
| 111 |
per_device_eval_batch_size=batch_size,
|
|
|
|
| 172 |
|
| 173 |
|
| 174 |
print("Save to Space")
|
| 175 |
+
#trainer.save_model("alexkueck/test-tis-1")
|
| 176 |
print("done")
|
| 177 |
|
| 178 |
#####################################
|
| 179 |
#Push to Hub
|
| 180 |
print("push to hub")
|
| 181 |
login(token=os.environ["HF_ACCESS_TOKEN"])
|
| 182 |
+
trainer.push_to_hub("alexkueck/test-tis-1")
|
| 183 |
print("done")
|
| 184 |
|
| 185 |
|