From 8b6db0c72d0991d3f474c09cd57820172c72225e Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Fri, 14 Feb 2025 20:11:31 +0700 Subject: [PATCH] fix: update default max_steps --- src/axolotl/core/trainer_builder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py index 09fbf077d..4517fe102 100755 --- a/src/axolotl/core/trainer_builder.py +++ b/src/axolotl/core/trainer_builder.py @@ -333,7 +333,7 @@ class TrainerBuilderBase(abc.ABC): self.cfg.save_total_limit if self.cfg.save_total_limit else 4 ) - training_args_kwargs["max_steps"] = self.cfg.max_steps or total_num_steps + training_args_kwargs["max_steps"] = self.cfg.max_steps or total_num_steps or -1 # max_length is not used in CausalTrainer if self.cfg.reward_model or self.cfg.rl: