src/diffusers/schedulers/scheduling_edm_euler.py · 9d0616189e67db6a56ad80737b30bfcde4ff877f · renzhc / diffusers_dcu

Slight performance improvement to `Euler`, `EDMEuler`, `FlowMatchHeun`, `KDPM2Ancestral` (#9616) · 9d061618

hlky authored Oct 15, 2024



* Slight performance improvement to Euler

* Slight performance improvement to EDMEuler

* Slight performance improvement to FlowMatchHeun

* Slight performance improvement to KDPM2Ancestral

* Update KDPM2AncestralDiscreteSchedulerTest

---------
Co-authored-by: YiYi Xu <yixu310@gmail.com>

9d061618

scheduling_edm_euler.py 16.9 KB

Replace scheduling_edm_euler.py