tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Set push to hub as private by default

Browse Source
This commit is contained in:
NanoCode012
2023-07-14 13:17:49 +09:00
committed by GitHub
parent ef17e15483
commit 1514739f0f
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/axolotl/utils/trainer.py
View File

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