diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py index bd9c30dda..09fbf077d 100755 --- a/src/axolotl/core/trainer_builder.py +++ b/src/axolotl/core/trainer_builder.py @@ -849,9 +849,6 @@ class HFCausalTrainerBuilder(TrainerBuilderBase): # https://docs.nvidia.com/deeplearning/performance/dl-performance-matrix-multiplication/index.html data_collator_kwargs["pad_to_multiple_of"] = multiple - if self.cfg.reward_model: - data_collator_kwargs["max_length"] = self.cfg.sequence_len - trainer_cls = self._get_trainer_cls() trainer_kwargs, trainer_cls = self.hook_pre_create_trainer( trainer_kwargs, trainer_cls @@ -931,8 +928,6 @@ class HFCausalTrainerBuilder(TrainerBuilderBase): collator_args = [self.tokenizer] if self.cfg.reward_model: collator = RewardDataCollatorWithPadding - if "max_length" in kwargs: - kwargs.pop("max_length") elif use_batch_sampler_collator: if self.cfg.flex_attention: collator = V2BatchSamplerDataCollatorForSeq2Seq