transformers

[tests] make `test_model_parallelism` device-agnostic

#30844

Merged