Created byWenliang Zhao*,Lujia Bai*,Yongming Rao,Jie Zhou,Jiwen Lu
This code contains the Pytorch implementation for UniPC (NeurIPS 2023).
Anonline demofor UniPC with stable-diffusion. Many thanks for the help and hardware resources supported by HuggingFace 🤗.
UniPC is a training-free framework designed for the fast sampling of diffusion models, which consists of a corrector (UniC) and a predictor (UniP) that share a unified analytical form and support arbitrary orders.
- 2023.3.11:UniPC is used instable-diffusion-webui.
- 2023.2.20:UniPC is integrated into 🧨 Diffusers and accesible via theUniPCMultistepScheduler.
UniPC is by designed model-agnostic, supporting pixel-space/latent-space DPMs on unconditional/conditional sampling. It can also be applied to both noise prediction model and data prediction model.
Compared with previous methods, UniPC converges faster thanks to the increased order of accuracy. Both quantitative and qualitative results show UniPC can remarkably improve the sampling quality, especially in extreme few steps (5~10).
We provide code examples based on theScoreSDEandStable-Diffusionin theexample
folder. Please follow theREADME.md
file in the corresponding examples for further instructions to use our UniPC.
We provide a pytorch example inexample/score_sde_pytorch
,where we show how to use our UniPC to sample from a DPM pre-trained on CIFAR10.
We provide an example of applying UniPC to stable-diffusion inexample/stable-diffusion
.Our UniPC can accelerate the sampling in both conditional and unconditional sampling.
UniPC is now also available in 🧨 Diffusers and accesible via theUniPCMultistepScheduler. Diffusers allows you to test UniPC in PyTorch in just a couple lines of code.
You can install diffusers as follows:
pip install diffusers accelerate transformers
And then try out the model with just a couple lines of code:
fromdiffusersimportStableDiffusionPipeline,UniPCMultistepScheduler
importtorch
path="CompVis/stable-diffusion-v1-4"
pipe=StableDiffusionPipeline.from_pretrained(path,torch_dtype=torch.float16)
# change to UniPC scheduler
pipe.scheduler=UniPCMultistepScheduler.from_config(pipe.scheduler.config)
pipe=pipe.to("cuda")
prompt="a highly realistic photo of green turtle"
generator=torch.manual_seed(0)
# only 15 steps are needed for good results => 2-4 seconds on GPU
image=pipe(prompt,generator=generator,num_inference_steps=15).images[0]
# save image
image.save("turtle.png")
For more information about UniPC anddiffusers
,please have a lookhereandhere.
Our code is based onScoreSDE,Stable-Diffusion,andDPM-Solver.
If you find our work useful in your research, please consider citing:
@article{zhao2023unipc,
title={UniPC: A Unified Predictor-Corrector Framework for Fast Sampling of Diffusion Models},
author={Zhao, Wenliang and Bai, Lujia and Rao, Yongming and Zhou, Jie and Lu, Jiwen},
journal={NeurIPS},
year={2023}
}