andreas128 / RePaint

Official PyTorch Code and Models of "RePaint: Inpainting using Denoising Diffusion Probabilistic Models", CVPR 2022

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Speed of Diffusion probabilistic models

ofirkris opened this issue · comments

Great work - results look very impressive.
From my experience Diffusion Models are quite heavy in terms of computation, can you share info about the inference speed?

Also interested, looks it is too slow to really use it.

@world2vec and @ofirkris If you think diffusion models are too slow you might want to check out this ICLR2022 submission then
https://openreview.net/pdf?id=TIdIXIpzhoI from Google Brain team.

Thanks a lot! Speed is a major issue, we put some ideas together to increase the speed of our inference schedule https://github.com/andreas128/RePaint#adapt-the-code

Does that help?