transformers

03462875 - Introduce `PartialState` as the device handler in the `Trainer` (#22752)

Introduce `PartialState` as the device handler in the `Trainer` (#22752) * Use accelerate for device management * Add accelerate to setup Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>