transformers

Trainer: make checkpoints work under tensor parallelism

#48569

Open