TheLastBen / fast-stable-diffusion

fast-stable-diffusion + DreamBooth
MIT License
7.49k stars 1.3k forks source link

DreamBooth: Option to add custom model paths insted of always download stable diffusion vanilla #109

Open esdrico opened 1 year ago

esdrico commented 1 year ago

hi, I love to use your colab, I would like to know if it would be possible to have an option to add custom models to train, (Like Waifu Diffusion for exemple)

TheLastBen commented 1 year ago

I'll add that option, but you'll need a diffusers model for training not a CKPT, but there are easy ways to convert them

ciaobnsr commented 1 year ago

would be very interesting to start from a different model. maybe you could include the conversion script from ckpt to diffusers.