tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: update default max_steps

Browse Source
This commit is contained in:
NanoCode012
2025-02-14 20:11:31 +07:00
parent 43f5373c79
commit 8b6db0c72d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/core/trainer_builder.py
View File

@@ -333,7 +333,7 @@ class TrainerBuilderBase(abc.ABC):
self.cfg.save_total_limit if self.cfg.save_total_limit else 4 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 # max_length is not used in CausalTrainer
if self.cfg.reward_model or self.cfg.rl: if self.cfg.reward_model or self.cfg.rl: