Dreambooth-Stable-Diffusion icon indicating copy to clipboard operation
Dreambooth-Stable-Diffusion copied to clipboard

Multiple persons

Open K4CZP3R opened this issue 1 year ago • 6 comments

Is it possible to train a model containing two persons?

K4CZP3R avatar Oct 04 '22 15:10 K4CZP3R

I would like to know the answer to this too.

I currently have 8 different 2GB model files, which all work really nicely. I wonder if it's possible to combine them and not risk losing data for any of them.

olinorwell avatar Oct 04 '22 18:10 olinorwell

I think, if you don't prune the new ckpt files, you can just train with one of those as your next --actual_resume path, since the unpruned one contains all of the master weight ckpt files data.

Gonna do some tests and see!

eadnams22 avatar Oct 05 '22 03:10 eadnams22

I tried that but the people became increasingly distorted with each new training/person added.

I'm trying this approach now https://github.com/XavierXiao/Dreambooth-Stable-Diffusion/issues/12 but that so far still ends up with just one person in the model.

thegenerativegeneration avatar Oct 05 '22 23:10 thegenerativegeneration

Hmm, yeah, I venture they each need to be in different classes, and not over trained? I'm still experimenting too, but noticing some distortions to the classes over time

eadnams22 avatar Oct 06 '22 17:10 eadnams22

Hmm, yeah, I venture they each need to be in different classes, and not over trained? I'm still experimenting too, but noticing some distortions to the classes over time

Eventually, do us the favour of telling the results of your experiments, pls 🙌

Jorvan758 avatar Oct 09 '22 00:10 Jorvan758

Seems like diverse enough classes seem to be okay, but man/woman bleed into each other, perhaps all that are close to each other will bleed into nearby classes?

I'd love a list of the noun classes, or a tree of their structure really, to better understand what it might be doing.

eadnams22 avatar Oct 09 '22 16:10 eadnams22

I actually tried this for two subjects with one class name describing two people in a set of images (eg: a photo of persons xx_yy) . After training the model is able to separate 'xx' and 'yy' as separate entities. Eg 'Male Person xx' and 'female person yy' will produce the proper outputs while still retaining the couple aspect. I didn't observe class bleeding but it could be just my experience.

maderix avatar Nov 10 '22 15:11 maderix

See this discord channel for more discussion on multiple subjects: https://discord.com/channels/1023277529424986162/1034634875782823997

djbielejeski avatar Feb 14 '23 21:02 djbielejeski