diffusionbee-stable-diffusion-ui icon indicating copy to clipboard operation
diffusionbee-stable-diffusion-ui copied to clipboard
verified publisher icon divamgupta

Support loading custom model checkpoints (ie DreamBooth)

Open uglyrobot opened this issue 3 years ago • 0 comments

I have a trained SD dreambooth model checkpoint. Tried to replace the default downloaded diffusion_model.h5 with it but it will just get overwritten when the downloader md5 checksum fails.

I think it would be simple to support this, maybe just look for a .ckpt file in ~/.diffusionbee/downloads and load that instead when present. That's how the other SD UIs work, I can do it with CHARL-E for mac as well.

That's all it would take to support DreamBooth for your users!

Awesome app BTW!

uglyrobot avatar Oct 06 '22 22:10 uglyrobot