transformers

43125: Ensure correct checkpoint saving behavior by simplifying Trainer.save_model parallelism logic

#43314

Merged