divamgupta / divamgupta/diffusionbee-stable-diffusion-ui
Support loading custom model checkpoints (ie DreamBooth)
Open
- Dominant language
- JavaScript
- Stars
- 13.6k
- Forks
- 727
- PR merge metrics
- No merged PRs in 30d
Description
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!
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.