transformers

Bring back `set_epoch` for Accelerate-based dataloaders

#26850

Merged