winglian commited on
Commit
008505c
1 Parent(s): b3f5e00

fix comma, not a tuple (#436)

Browse files
Files changed (1) hide show
  1. src/axolotl/utils/trainer.py +1 -1
src/axolotl/utils/trainer.py CHANGED
@@ -461,7 +461,7 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer, total_num_
461
  training_arguments_kwargs["save_strategy"] = cfg.save_strategy
462
  else:
463
  training_arguments_kwargs["save_strategy"] = (
464
- "steps" if cfg.save_steps else "epoch",
465
  )
466
 
467
  training_args = AxolotlTrainingArguments( # pylint: disable=unexpected-keyword-arg
 
461
  training_arguments_kwargs["save_strategy"] = cfg.save_strategy
462
  else:
463
  training_arguments_kwargs["save_strategy"] = (
464
+ "steps" if cfg.save_steps else "epoch"
465
  )
466
 
467
  training_args = AxolotlTrainingArguments( # pylint: disable=unexpected-keyword-arg