ShadeEngine's picture
End of training
4ac8f3e verified

KDPM2DiscreteScheduler

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.

The original codebase can be found at crowsonkb/k-diffusion.

KDPM2DiscreteScheduler

[[autodoc]] KDPM2DiscreteScheduler

SchedulerOutput

[[autodoc]] schedulers.scheduling_utils.SchedulerOutput