fix red CI

huggingface · Jun 17, 2024 · 4e23d95 · 4e23d95
1 parent 50c4620
commit 4e23d95
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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__()
diff --git 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__()