tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix comma, not a tuple (#436)

Browse Source
This commit is contained in:
Wing Lian
2023-08-19 00:57:40 -04:00
committed by GitHub
parent b3f5e00ff5
commit 008505c8ae
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/trainer.py
View File

@@ -461,7 +461,7 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer, total_num_
training_arguments_kwargs["save_strategy"] = cfg.save_strategy training_arguments_kwargs["save_strategy"] = cfg.save_strategy
else: else:
training_arguments_kwargs["save_strategy"] = ( training_arguments_kwargs["save_strategy"] = (
"steps" if cfg.save_steps else "epoch", "steps" if cfg.save_steps else "epoch"
) )
training_args = AxolotlTrainingArguments( # pylint: disable=unexpected-keyword-arg training_args = AxolotlTrainingArguments( # pylint: disable=unexpected-keyword-arg