transformers

4db82d46 - Optimize the device allocation in case of hidden_states in multiple devices