Enable naive Pipeline Parallelism training for Gpt neox japanese and san japanese
As suggested in the https://github.com/huggingface/transformers/issues/22561, moved labels to the same device as logits for GPT Neox Japanese and Gpt San Japanese
@sgugger Could you review this once?<|||||>_The documentation is not available anymore as the PR was closed or merged._