Update cross_entropy_loss.py

2024-02-26 03:09:37 +11:00 · 2024-02-26 03:09:37 +11:00 · fdaa9493be
commit fdaa9493be
parent 8259b16a5d
1 changed files with 0 additions and 1 deletions
--- a/unsloth/kernels/cross_entropy_loss.py
+++ b/unsloth/kernels/cross_entropy_loss.py
@ -217,7 +217,6 @@ class Fast_CrossEntropyLoss(torch.autograd.Function):
                losses,
                logsumexp,
                labels,
-                n_rows,
                VOCAB_SIZE = vocab_size,
                BLOCK_SIZE = MAX_FUSED_SIZE,
                num_warps  = 32,