formatting

linkedin · Dec 13, 2024 · ab08ab6 · ab08ab6
1 parent d3f565b
commit ab08ab6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/liger_kernel/chunked_loss/kto_loss.py b/src/liger_kernel/chunked_loss/kto_loss.py
@@ -59,7 +59,11 @@ def preference_loss_fn(
         chosen_rewards = beta * chosen_logratios.detach()
         rejected_rewards = beta * rejected_logratios.detach()
 
-        return losses.sum() / (full_target.shape[0] // 2), chosen_rewards, rejected_rewards
+        return (
+            losses.sum() / (full_target.shape[0] // 2),
+            chosen_rewards,
+            rejected_rewards,
+        )
 
     @staticmethod
     def forward(