#114 · Primary category: Image Generation
dpm-solver
Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)
Project last updated:02/06/24
GitHub Stars
1.9K
Forks
136
Contributors
4
License
MIT
Why we included this project
Diffusion sampling is slow because it normally needs dozens of denoising steps, and DPM-Solver attacks exactly that bottleneck. This repo implements the DPM-Solver and DPM-Solver++ algorithms from the NeurIPS 2022 papers as high-order ODE solvers that bring the needed function evaluations down to around 10 to 20, with no retraining. Because it is model-agnostic and training-free, it slots in as a drop-in replacement sampler for discrete-time and continuous-time diffusion models alike; it also underlies the default multistep scheduler in Hugging Face Diffusers for Stable Diffusion pipelines. If you run guided-diffusion workloads such as text-to-image or inpainting on existing models and want them to feel snappier, this gives you a direct way to get that speed. The repo also includes working integration examples for Stable Diffusion, DeepFloyd-IF, and guided diffusion, so it works both as a solver to adopt and as a reference for wiring advanced samplers into real pipelines.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
stable-diffusion-webui
Stable Diffusion web UI
ComfyUI
The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface.
diffusers
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
upscayl
🆙 Upscayl - #1 Free and Open Source AI Image Upscaler for Linux, MacOS and Windows.
InvokeAI
InvokeAI is a leading open-source creative engine for Stable Diffusion, offering an industry-leading web UI for generating and editing visual media.