dpm-solver

收录理由

扩散模型采样通常需要数十步去噪，速度很慢，DPM-Solver 正是针对这一瓶颈设计的。这个仓库实现了 NeurIPS 2022 论文中的 DPM-Solver 和 DPM-Solver++ 算法，作为高阶 ODE 求解器，能把所需的函数评估次数压缩到大约 10 到 20 次，而且无需重新训练。由于它不依赖特定模型结构，也不需要训练，所以可以直接作为离散时间或连续时间扩散模型的采样器替换使用；同时，它也是 Hugging Face Diffusers 中 Stable Diffusion 管道默认多步调度器的基础。如果你在现有模型上运行引导扩散任务，比如文生图或图像修复，并希望让生成过程更快捷，这个工具提供了一条直接的提速路径。仓库里还包含了针对 Stable Diffusion、DeepFloyd-IF 和 guided diffusion 的可用集成示例，既可以作为求解器直接采用，也能作为把高级采样器接入真实管线的参考实现。