transformers

Fix failing `test_batch_generation` for bloom

#25718

Merged