Merge branch 'OpenAccess-AI-Collective:main' into logging_enhancement

This commit is contained in:
The Objective Dad
2023-07-15 06:16:04 -05:00
committed by GitHub
2 changed files with 9 additions and 0 deletions

View File

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