transformers

3378e7da - Move num_items_in_batch to correct device before accelerator.gather (#40773)