From 8f8e95e25d10c433cc1f2f8c7dcfed218bb13ac7 Mon Sep 17 00:00:00 2001 From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> Date: Mon, 17 Jun 2024 16:49:00 +0200 Subject: [PATCH] CPO / DPO: Fix red CI (#1749) * fix red CI * precommit --- trl/trainer/cpo_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/trl/trainer/cpo_config.py b/trl/trainer/cpo_config.py index 99c74e2c87..e5a9e4b7ba 100644 --- 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__() \ No newline at end of file + return super().__post_init__()