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
|