MuseV-test / diffusers /docs /source /en /api /schedulers /singlestep_dpm_solver.md
kevinwang676's picture
Upload folder using huggingface_hub
6755a2d verified

A newer version of the Gradio SDK is available: 5.16.0

Upgrade

DPMSolverSinglestepScheduler

DPMSolverSinglestepScheduler is a single step scheduler from DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps and DPM-Solver++: Fast Solver for Guided Sampling of Diffusion Probabilistic Models by Cheng Lu, Yuhao Zhou, Fan Bao, Jianfei Chen, Chongxuan Li, and Jun Zhu.

DPMSolver (and the improved version DPMSolver++) is a fast dedicated high-order solver for diffusion ODEs with convergence order guarantee. Empirically, DPMSolver sampling with only 20 steps can generate high-quality samples, and it can generate quite good samples even in 10 steps.

The original implementation can be found at LuChengTHU/dpm-solver.

Tips

It is recommended to set solver_order to 2 for guide sampling, and solver_order=3 for unconditional sampling.

Dynamic thresholding from Imagen is supported, and for pixel-space diffusion models, you can set both algorithm_type="dpmsolver++" and thresholding=True to use dynamic thresholding. This thresholding method is unsuitable for latent-space diffusion models such as Stable Diffusion.

DPMSolverSinglestepScheduler

[[autodoc]] DPMSolverSinglestepScheduler

SchedulerOutput

[[autodoc]] schedulers.scheduling_utils.SchedulerOutput