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
|