diff --git a/trl/trainer/dpo_trainer.py b/trl/trainer/dpo_trainer.py index fb79ef0e393..0ff1dfce04c 100644 --- a/trl/trainer/dpo_trainer.py +++ b/trl/trainer/dpo_trainer.py @@ -167,7 +167,7 @@ def __init__( ): if model_init_kwargs is not None: warnings.warn( - "You passed `model_init_kwargs` to the SFTTrainer, the value you passed will override the one in the `SFTConfig`." + "You passed `model_init_kwargs` to the DPOTrainer, the value you passed will override the one in the `DPOConfig`." ) args.model_init_kwargs = model_init_kwargs @@ -187,7 +187,7 @@ def __init__( if ref_model_init_kwargs is not None: warnings.warn( - "You passed `ref_model_kwargs` to the SFTTrainer, the value you passed will override the one in the `SFTConfig`." + "You passed `ref_model_init_kwargs` to the DPOTrainer, the value you passed will override the one in the `DPOConfig`." ) args.ref_model_init_kwargs = ref_model_init_kwargs