Nanobit commited on
Commit
168a7a0
2 Parent(s): 231031a 1514739

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

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
  if cfg.save_safetensors:
186
  training_arguments_kwargs["save_safetensors"] = cfg.save_safetensors
 
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
  if cfg.save_safetensors:
187
  training_arguments_kwargs["save_safetensors"] = cfg.save_safetensors