Merge pull request #274 from OpenAccess-AI-Collective/NanoCode012-patch-2

Feat: Set push to hub as private by default
This commit is contained in:
NanoCode012
2023-07-14 23:15:47 +09:00
committed by GitHub

View File

@@ -181,6 +181,7 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer):
if cfg.hub_model_id:
training_arguments_kwargs["hub_model_id"] = cfg.hub_model_id
training_arguments_kwargs["push_to_hub"] = True
training_arguments_kwargs["hub_private_repo"] = True
if cfg.save_safetensors:
training_arguments_kwargs["save_safetensors"] = cfg.save_safetensors