[train_dreambooth_lora_sdxl.py] Fix the LR Schedulers when num_train_epochs is passed in a distributed training env#11240

sayakpaul merged 4 commits intohuggingface:mainhuggingface/diffusers:mainfrom

kghamilton89:dreambooth-lora-sdxl-num-train-epochs-patchkghamilton89/diffusers:dreambooth-lora-sdxl-num-train-epochs-patchCopy head branch name to clipboard