fix comma, not a tuple (#436)

2023-08-19 00:57:40 -04:00
parent b3f5e00ff5
commit 008505c8ae
1 changed files with 1 additions and 1 deletions
--- 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