transformers

78aee7e8 - Tensor Parallelism and `mps` device (#44506)

Tensor Parallelism and `mps` device (#44506) fix: fail when using `mps` with TP Co-authored-by: Ferdinand Mom <47445085+3outeille@users.noreply.github.com>