Merge pull request #274 from OpenAccess-AI-Collective/NanoCode012-patch-2
Browse files
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
|