diff --git a/trl/trainer/cpo_config.py b/trl/trainer/cpo_config.py index d231fa83cc..99c74e2c87 100644 --- a/trl/trainer/cpo_config.py +++ b/trl/trainer/cpo_config.py @@ -83,3 +83,4 @@ class CPOConfig(TrainingArguments): def __post_init__(self): if self.loss_type == "kto_pair": raise ValueError("Support for kto_pair has been removed in CPOTrainer. Please use KTOTrainer.") + return super().__post_init__() \ No newline at end of file diff --git a/trl/trainer/dpo_config.py b/trl/trainer/dpo_config.py index 88b8b2d1fb..38857409b0 100644 --- a/trl/trainer/dpo_config.py +++ b/trl/trainer/dpo_config.py @@ -106,3 +106,4 @@ class DPOConfig(TrainingArguments): def __post_init__(self): if self.loss_type == "kto_pair": raise ValueError("Support for kto_pair has been removed in DPOTrainer. Please use KTOTrainer.") + return super().__post_init__()