src/diffusers/schedulers/scheduling_heun_discrete.py · b934215d4c376ea2e08e28103443686b95ea772c · renzhc / diffusers_dcu

[scheduler] support custom `timesteps` and `sigmas` (#7817) · b934215d

YiYi Xu authored May 09, 2024



* support custom sigmas and timesteps, dpm euler

---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
Co-authored-by: Benjamin Bossan <BenjaminBossan@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

b934215d

scheduling_heun_discrete.py 21.9 KB

Replace scheduling_heun_discrete.py