fix save_steps so it doesn't get duplicated (axolotl-ai-cloud#567)

mkeoliya · Sep 14, 2023 · 72e96f2 · 72e96f2
1 parent 5f8d74f
commit 72e96f2
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/axolotl/utils/trainer.py b/src/axolotl/utils/trainer.py
@@ -640,7 +640,6 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer, total_num_
         eval_accumulation_steps=cfg.gradient_accumulation_steps,
         num_train_epochs=cfg.num_epochs,
         learning_rate=cfg.learning_rate,
-        save_steps=cfg.save_steps,
         output_dir=cfg.output_dir,
         save_total_limit=cfg.save_total_limit if cfg.save_total_limit else 4,
         load_best_model_at_end=(