43125: Ensure correct checkpoint saving behavior by simplifying Trainer.save_model parallelism logic (#43314)
* 43125: Ensure correct checkpoint saving behavior by simplifying Trainer.save_model parallelism logic
* removed tp_size check in trainer
---------
Co-authored-by: Kashif Rasul <kashif.rasul@gmail.com>