Skip to content

DPMSolverSinglestepScheduler

DPMSolverSinglestepScheduler 是来自 DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 StepsDPM-Solver++: Fast Solver for Guided Sampling of Diffusion Probabilistic Models 的单步调度器,由 Cheng Lu、Yuhao Zhou、Fan Bao、Jianfei Chen、Chongxuan Li 和 Jun Zhu 开发。

DPMSolver(及其改进版本 DPMSolver++)是一种快速专用的高阶扩散 ODE 求解器,具有收敛阶保证。在经验上,DPMSolver 仅用 20 步即可生成高质量的样本,甚至在 10 步内也能生成相当好的样本。

原始实现可以在 LuChengTHU/dpm-solver 找到。

提示

建议在引导采样时将 solver_order 设置为 2,在无条件采样时设置为 solver_order=3

支持来自 Imagen 的动态阈值化,对于像素空间扩散模型,你可以同时设置 algorithm_type="dpmsolver++"thresholding=True 以使用动态阈值化。这种方法不适用于潜在空间扩散模型,如 Stable Diffusion。

DPMSolverSinglestepScheduler

[[autodoc]] DPMSolverSinglestepScheduler

SchedulerOutput

[[autodoc]] schedulers.scheduling_utils.SchedulerOutput