AITemplate icon indicating copy to clipboard operation
AITemplate copied to clipboard

Published 20 hours ago • verified publisher icon facebookincubator

CompVis/Stable Diffusion/LDM - Diffusion Models Implementation Support

Open Ahmer-444 opened this issue 2 years ago • 2 comments

Hi!

Have anyone worked or planned to integrate diffusion models from CompVis/stable-diffusion/ldm? Would be of great help as of my experience that implementation is working quite good on characters category. As they have mentioned, used one (PLMS) from Katherine Crowson's Implementation.

Thanks

Ahmer-444 avatar Oct 12 '22 01:10 Ahmer-444

You can directly use call these sampler from pytorch: https://github.com/facebookincubator/AITemplate/blob/main/examples/05_stable_diffusion/pipeline_stable_diffusion_ait.py#L74

terrychenism avatar Oct 12 '22 17:10 terrychenism

What @Ahmer-444 says is that the samplers like DPM2 a and Euler a can't be used as samplers.

flipkast avatar Oct 13 '22 15:10 flipkast