How to reduce the size of trained model ?

[sausax]

Any ideas on how to reduce the size of the trained model from 12gb to 4gb(same as original stable diffusion checkpoint) ?

[kitty7779]

I used this script which reduced it to 2gb in my case. https://github.com/harubaru/waifu-diffusion/blob/main/scripts/prune.py Replace "models/ldm/stable-diffusion-v1/model.ckpt" with the path to whatever ckpt file you want to prune

[JoePenna]

+1 on what @kitty7779 said.

This repo automatically prunes the ckpt down to 2GB after creation (takes about 15-20 extra seconds) https://github.com/JoePenna/Dreambooth-Stable-Diffusion/

Use main_prune.py for that.