feat(model parallelism): moving the labels to the same device as the logits for gpt2 and bart#22591

sgugger merged 2 commits into

huggingface:mainhuggingface/transformers:mainfrom

kausmeows:kausCopy head branch name to clipboard