Use torch.Tensor.reshape for non-contiguous tensor in ce loss function#591

danielhanchen merged 2 commits into

unslothai:mainunslothai/unsloth-zoo:mainfrom

noonamkha:fix/ce-loss-non-contiguous-tensornoonamkha/unsloth-zoo:fix/ce-loss-non-contiguous-tensorCopy head branch name to clipboard