Dreambooth Tab

[heartedtran]

Hello! I think it would be cool to have a tab to train a dream booth model, not sure if it's something being worked on right now!

https://github.com/XavierXiao/Dreambooth-Stable-Diffusion

It seems to be based on the Textual Inversion repo as well

[ProGamerGov]

I've found this repo works pretty well (The config used for the RunPod version, as the Colab version has subpar quality): https://github.com/JoePenna/Dreambooth-Stable-Diffusion, though its epoch / step counting is a bit weird. The XavierXiao requires using the word 'sks', which seems to be because of a unresolved mistake.

[ClashSAN]

@heartedtran yes, plenty of requests already. there is also a PR with someone working o this. Exciting :)
#1734 #1280

[heartedtran]

ohh i see, i wasn't able to find these with the search function for some reason

[0xdevalias]

Dreambooth is available as an extension now, see below:

Closing this, as I've now started a repo with a standalone extension based on ShivShiram's repo here:

https://github.com/d8ahazard/sd_dreambooth_extension

Please feel free to test and yell at me there. I've added requirements installer, multiple concept training via JSON, and moved some bit about.

UI still needs fixing, some stuff broken there, but it should be able to train a model for now.

Originally posted by @d8ahazard in #3995 (comment)