Format

Signed-off-by: Austin Liu <[email protected]>

Author: austin362667

src/liger_kernel/chunked_loss/dpo_loss.py

@@ -99,6 +99,7 @@ def backward(ctx, grad_output):
         # Return these gradients, followed by None for the remaining inputs
         return d_input, d_weight, d_target, None, None, d_bias, None, None, None
 
+
 class LigerFusedLinearDPOLoss(torch.nn.Module):
     """
     Fused linear layer with DPO loss.
@@ -132,4 +133,4 @@ def forward(self, lin_weight, _input, target, bias=None):
             self.beta,
             self.compute_nll_loss,
             self.compiled,
-        )
+        )