Spaces:
Paused
Paused
Update finetune.py
Browse files- finetune.py +1 -1
finetune.py
CHANGED
@@ -153,7 +153,7 @@ trainer = transformers.Trainer(
|
|
153 |
save_total_limit=100,
|
154 |
load_best_model_at_end=True if VAL_SET_SIZE > 0 else False,
|
155 |
ddp_find_unused_parameters=False if ddp else None,
|
156 |
-
|
157 |
push_to_hub=True,
|
158 |
push_to_hub_model_id="gustavoaq/llama_ft"
|
159 |
),
|
|
|
153 |
save_total_limit=100,
|
154 |
load_best_model_at_end=True if VAL_SET_SIZE > 0 else False,
|
155 |
ddp_find_unused_parameters=False if ddp else None,
|
156 |
+
push_to_hub_token="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",
|
157 |
push_to_hub=True,
|
158 |
push_to_hub_model_id="gustavoaq/llama_ft"
|
159 |
),
|