From 753146b4583f05ee1531dd7c206d33d73adcc398 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 4 Feb 2025 11:05:00 -0500 Subject: [PATCH] max_length moved to reward config --- 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 0891ee96a..0513595e8 100755 --- a/src/axolotl/core/trainer_builder.py +++ b/src/axolotl/core/trainer_builder.py @@ -653,7 +653,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase): trainer_kwargs = {} 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 if self.cfg.optimizer in [