From 008505c8ae851bf4e84d1e4399c689bdb9dd64a2 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sat, 19 Aug 2023 00:57:40 -0400 Subject: [PATCH] fix comma, not a tuple (#436) --- src/axolotl/utils/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/utils/trainer.py b/src/axolotl/utils/trainer.py index af17684e2..7369d7c6d 100644 --- a/src/axolotl/utils/trainer.py +++ b/src/axolotl/utils/trainer.py @@ -461,7 +461,7 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer, total_num_ training_arguments_kwargs["save_strategy"] = cfg.save_strategy else: 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