[train_instruct_pix2pix.py]Fix the LR schedulers when `num_train_epochs` is passed in a distributed training env#9316

sayakpaul merged 5 commits intohuggingface:mainhuggingface/diffusers:mainfrom

AnandK27:AnandK27-patch-pix2pix-lr-schedulerAnandK27/diffusers:AnandK27-patch-pix2pix-lr-schedulerCopy head branch name to clipboard