madebyollin / maple-diffusion

Stable Diffusion inference on iOS / macOS using MPSGraph
https://madebyoll.in/posts/maple_diffusion/
MIT License
793 stars 51 forks source link

How to obtain the `.ckpt` file from original stable diffusion model? #30

Closed geekinglcq closed 1 year ago

geekinglcq commented 1 year ago

Hi, thank you for your amazing job. I am trying to convert the stable diffusion model but I met some problems.
I have found that the original stable diffusion model consists of many files, configs and model bins, but here the convert script seems only receive single file as input. So, How to obtain the single .ckpt file from the original stable diffusion model? It's ok to save one of the module's state_dict such as unet, but how about other modules?

madebyollin commented 1 year ago

I've corrected the v1.4 link in the README to point to https://huggingface.co/CompVis/stable-diffusion-v-1-4-original, which should have the single .ckpt file available.

geekinglcq commented 1 year ago

Thanks~ Problem solved.