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

max_length moved to reward config

Browse Source
This commit is contained in:
Wing Lian
2025-02-04 11:05:00 -05:00
parent d683c50113
commit 753146b458
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

@@ -653,7 +653,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
trainer_kwargs = {} trainer_kwargs = {}
if self.cfg.reward_model: if self.cfg.reward_model:
trainer_kwargs["max_length"] = self.cfg.sequence_len training_arguments_kwargs["max_length"] = self.cfg.sequence_len
# pylint: disable=duplicate-code # pylint: disable=duplicate-code
if self.cfg.optimizer in [ if self.cfg.optimizer in [