Aminrabi's picture
End of training
c0af20c
|
raw
history blame
994 Bytes

DPM Stochastic Scheduler inspired by Karras et. al paper

Overview

Inspired by Stochastic Sampler from Karras et. al. Scheduler ported from @crowsonkb's https://github.com/crowsonkb/k-diffusion library:

All credit for making this scheduler work goes to Katherine Crowson

DPMSolverSDEScheduler

[[autodoc]] DPMSolverSDEScheduler