Skip to content

DPMSolverMultistepScheduler

DPMSolverMultistepScheduler 是来自 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 步内也能生成相当好的样本。

提示

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

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

DPMSolver 和 DPM-Solver++ 的 SDE 变体也得到支持,但仅适用于一阶和二阶求解器。这是一种用于反向扩散 SDE 的快速 SDE 求解器。建议使用二阶的 sde-dpmsolver++

DPMSolverMultistepScheduler

[[autodoc]] DPMSolverMultistepScheduler

SchedulerOutput

[[autodoc]] schedulers.scheduling_utils.SchedulerOutput