Skip to content

Fix ZeroDivisionError in fused cross entropy when GPU memory exhausted#432

Merged
danielhanchen merged 2 commits into
unslothai:mainfrom
GabrielArpini:fix-zerodivision-fusedce
Jan 4, 2026
Merged

Fix ZeroDivisionError in fused cross entropy when GPU memory exhausted#432
danielhanchen merged 2 commits into
unslothai:mainfrom
GabrielArpini:fix-zerodivision-fusedce

Update unsloth_zoo/fused_losses/cross_entropy_loss.py

a55c253
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs