wangjin2000 commited on
Commit
5b50c1d
·
verified ·
1 Parent(s): 53cd821

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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