算力平台:
DPMSolverSinglestepScheduler
DPMSolverSinglestepScheduler
是来自 DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps 和 DPM-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