transformers

Fix import for torch 2.0, 2.1 - guard typehint for "device_mesh"

#36768

Merged