transformers

[docs] use device-agnostic API instead of hard-coded cuda

#35048

Merged