fix: update default max_steps

axolotl-ai-cloud · Feb 18, 2025 · 93a2ecc · 93a2ecc
1 parent 8b327a7
commit 93a2ecc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py
@@ -327,7 +327,7 @@ def _set_base_training_args(self, total_num_steps) -> dict[str, Any]:
             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: