The KDPM2DiscreteScheduler is inspired by the Elucidating the Design Space of Diffusion-Based Generative Models paper, and the scheduler is ported from and created by Katherine Crowson.
KDPM2DiscreteScheduler
The original codebase can be found at crowsonkb/k-diffusion.
[[autodoc]] KDPM2DiscreteScheduler
[[autodoc]] schedulers.scheduling_utils.SchedulerOutput