fix comma, not a tuple (#436)
Browse files
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
|