diffusers

8e3affc6 - fix for lr scheduler in distributed training (#9103)

fix for lr scheduler in distributed training (#9103) * fix for lr scheduler in distributed training * Fixed the recalculation of the total training step section * Fixed lint error --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>