TheLastBen / fast-stable-diffusion

fast-stable-diffusion + DreamBooth
MIT License
7.57k stars 1.31k forks source link

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

Open esdrico opened 2 years ago

esdrico commented 2 years 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 2 years 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 2 years ago

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