Skip to content

Fix: cast logits to float32 in cross_entropy_forward to prevent errors#1254

Merged
danielhanchen merged 2 commits into
unslothai:nightlyfrom
Erland366:fix/casting_error
Nov 6, 2024
Merged

Fix: cast logits to float32 in cross_entropy_forward to prevent errors#1254
danielhanchen merged 2 commits into
unslothai:nightlyfrom
Erland366:fix/casting_error

Commits