Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -187,7 +187,9 @@ def train_function_no_sweeps(base_model_path): #, train_dataset, test_dataset)
|
|
| 187 |
seed=8893,
|
| 188 |
fp16=True,
|
| 189 |
#report_to='wandb'
|
| 190 |
-
report_to=None
|
|
|
|
|
|
|
| 191 |
)
|
| 192 |
|
| 193 |
# Initialize Trainer
|
|
|
|
| 187 |
seed=8893,
|
| 188 |
fp16=True,
|
| 189 |
#report_to='wandb'
|
| 190 |
+
report_to=None,
|
| 191 |
+
push_to_hub=True, #jw 20240701
|
| 192 |
+
hub_token = access_token, #jw 20240701
|
| 193 |
)
|
| 194 |
|
| 195 |
# Initialize Trainer
|