Spaces:
Paused
Paused
Update finetune.py
Browse files- finetune.py +0 -3
finetune.py
CHANGED
@@ -153,9 +153,6 @@ 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 |
-
push_to_hub_token="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",
|
157 |
-
push_to_hub=True,
|
158 |
-
push_to_hub_model_id="llama_ft"
|
159 |
),
|
160 |
data_collator=transformers.DataCollatorForLanguageModeling(tokenizer, mlm=False),
|
161 |
)
|
|
|
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 |
data_collator=transformers.DataCollatorForLanguageModeling(tokenizer, mlm=False),
|
158 |
)
|