wangjin2000 commited on
Commit
4137ce7
·
verified ·
1 Parent(s): 5b50c1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -178,7 +178,7 @@ def train_function_no_sweeps(base_model_path): #, train_dataset, test_dataset)
178
  load_best_model_at_end=True,
179
  metric_for_best_model="f1",
180
  greater_is_better=True,
181
- push_to_hub=False,
182
  logging_dir=None,
183
  logging_first_step=False,
184
  logging_steps=200,
@@ -188,7 +188,6 @@ def train_function_no_sweeps(base_model_path): #, train_dataset, test_dataset)
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
 
@@ -212,6 +211,8 @@ def train_function_no_sweeps(base_model_path): #, train_dataset, test_dataset)
212
  return save_path
213
 
214
  # Constants & Globals
 
 
215
  MODEL_OPTIONS = [
216
  "facebook/esm2_t6_8M_UR50D",
217
  "facebook/esm2_t12_35M_UR50D",
 
178
  load_best_model_at_end=True,
179
  metric_for_best_model="f1",
180
  greater_is_better=True,
181
+ push_to_hub=True, #jw 20240701False,
182
  logging_dir=None,
183
  logging_first_step=False,
184
  logging_steps=200,
 
188
  fp16=True,
189
  #report_to='wandb'
190
  report_to=None,
 
191
  hub_token = access_token, #jw 20240701
192
  )
193
 
 
211
  return save_path
212
 
213
  # Constants & Globals
214
+ access_token = os.getenv("hf_access_tokens")
215
+
216
  MODEL_OPTIONS = [
217
  "facebook/esm2_t6_8M_UR50D",
218
  "facebook/esm2_t12_35M_UR50D",