CPO / DPO: Fix red CI (#1749)

* fix red CI * precommit
huggingface · Jun 17, 2024 · 8f8e95e · 8f8e95e
1 parent 4e23d95
commit 8f8e95e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trl/trainer/cpo_config.py b/trl/trainer/cpo_config.py
@@ -83,4 +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__()
+        return super().__post_init__()