transformers

66677262 - Specify device in DataCollator in order to place tensors directly on the right device and avoid moving data around afterwards.