Skip to content

Commit 168a7a0

Browse files
authored
Merge pull request axolotl-ai-cloud#274 from OpenAccess-AI-Collective/NanoCode012-patch-2
Feat: Set push to hub as private by default
2 parents 231031a + 1514739 commit 168a7a0

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Diff for: src/axolotl/utils/trainer.py

+1
Original file line numberDiff line numberDiff line change
@@ -181,6 +181,7 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer):
181181
if cfg.hub_model_id:
182182
training_arguments_kwargs["hub_model_id"] = cfg.hub_model_id
183183
training_arguments_kwargs["push_to_hub"] = True
184+
training_arguments_kwargs["hub_private_repo"] = True
184185

185186
if cfg.save_safetensors:
186187
training_arguments_kwargs["save_safetensors"] = cfg.save_safetensors

0 commit comments

Comments
 (0)