remove unused

linkedin · Dec 17, 2024 · 8273696 · 8273696
1 parent d8e457b
commit 8273696
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/src/liger_kernel/chunked_loss/fused_linear_preference.py b/src/liger_kernel/chunked_loss/fused_linear_preference.py
@@ -308,10 +308,6 @@ def chunk_forward(
                 reduction="sum",
                 ignore_index=ignore_index,
             )
-        else:
-            chosen_nll_loss = torch.zeros(
-                (), device=target_chunk.device, dtype=target_chunk.dtype
-            )
 
         loss_mask = target_chunk != ignore_index
         label_chunk = torch.where(loss_mask, target_chunk, 0)