transformers

Fix device issue in `OpenLlamaModelTest::test_model_parallelism`

#24195

Merged