Spaces:
Paused
Paused
Update app.py
Browse files
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=
|
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",
|