Feat(config): Add hub_strategy (#386)

This commit is contained in:
NanoCode012
2023-08-14 20:12:55 +09:00
committed by GitHub
parent 63fdb5a7fb
commit 73a0b6ead5
2 changed files with 6 additions and 0 deletions

View File

@@ -440,6 +440,9 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer, total_num_
training_arguments_kwargs["push_to_hub"] = True
training_arguments_kwargs["hub_private_repo"] = True
if cfg.hub_strategy:
training_arguments_kwargs["hub_strategy"] = cfg.hub_strategy
if cfg.save_safetensors:
training_arguments_kwargs["save_safetensors"] = cfg.save_safetensors