diffusers

fix(training): lr scheduler doesn't work properly in distributed scenarios

#8312

Merged