Nanobit commited on
Commit
1514739
1 Parent(s): ef17e15

Set push to hub as private by default

Browse files
Files changed (1) hide show
  1. src/axolotl/utils/trainer.py +1 -0
src/axolotl/utils/trainer.py CHANGED
@@ -181,6 +181,7 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer):
181
  if cfg.hub_model_id:
182
  training_arguments_kwargs["hub_model_id"] = cfg.hub_model_id
183
  training_arguments_kwargs["push_to_hub"] = True
 
184
 
185
  training_args = AxolotlTrainingArguments(
186
  per_device_train_batch_size=cfg.micro_batch_size,
 
181
  if cfg.hub_model_id:
182
  training_arguments_kwargs["hub_model_id"] = cfg.hub_model_id
183
  training_arguments_kwargs["push_to_hub"] = True
184
+ training_arguments_kwargs["hub_private_repo"] = True
185
 
186
  training_args = AxolotlTrainingArguments(
187
  per_device_train_batch_size=cfg.micro_batch_size,