The KDPM2DiscreteScheduler with ancestral sampling 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.
The original codebase can be found at crowsonkb/k-diffusion.
[[autodoc]] KDPM2AncestralDiscreteScheduler
[[autodoc]] schedulers.scheduling_utils.SchedulerOutput