diffusers

cc2c4ae7 - fix inference in custom diffusion (#5329)

fix inference in custom diffusion (#5329) * Update train_custom_diffusion.py * make style * Empty-Commit --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>