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",
|