[FIX] Move labels to logits device in cross-entropy loss for multi-GPU support (#4041)#4059

danielhanchen merged 1 commit into

unslothai:mainunslothai/unsloth:mainfrom

devchilll:fix/4041-cross-entropy-device-mismatchdevchilll/unsloth:fix/4041-cross-entropy-device-mismatchCopy head branch name to clipboard