Update cross_entropy_loss.py

2024-11-04 19:48:09 -08:00 · 2024-11-04 19:48:09 -08:00 · cc563fac52
commit cc563fac52
parent 97f37ace96
1 changed files with 0 additions and 2 deletions
--- a/unsloth/kernels/cross_entropy_loss.py
+++ b/unsloth/kernels/cross_entropy_loss.py
@ -348,8 +348,6 @@ class Fast_CrossEntropyLoss(torch.autograd.Function):
        div, mod = divmod(vocab_size, BLOCK_SIZE)
        n_blocks : int = div + (mod != 0)

-        print("111")
-
        _cross_entropy_backward[(n_rows, n_blocks,)](
            logits,   logits.stride(0),
            dlosses, dlosses.stride(0),