diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py index 6380e7186..d4aa1ddf1 100755 --- a/src/axolotl/core/trainer_builder.py +++ b/src/axolotl/core/trainer_builder.py @@ -1000,10 +1000,6 @@ class HFRLTrainerBuilder(TrainerBuilderBase): total_num_steps=total_num_steps ) - training_args_kwargs["lr_scheduler_type"] = ( - self.cfg.lr_scheduler if self.cfg.lr_scheduler else "cosine" - ) - if self.cfg.remove_unused_columns is not None: training_args_kwargs["remove_unused_columns"] = ( self.cfg.remove_unused_columns