sd-scripts icon indicating copy to clipboard operation
sd-scripts copied to clipboard

[SDXL] Train SDXL refiner for subject generation

Open jkcarney opened this issue 2 years ago • 9 comments

I've been using the scripts here to fine tune the base SDXL model for subject driven generation to good effect. However, I've found that adding the refiner step usually means that the refiner doesn't understand the subject, which often makes using the refiner worse with subject generation.

Currently the scripts do not support fine tuning the refiner to make generation better. When tuning the refiner, realistically we would only need to fine tune the text encoder, not the u-net. If this ability was added, I think that would be very beneficial.

jkcarney avatar Jul 03 '23 14:07 jkcarney

Yes this would be great

TingTingin avatar Jul 04 '23 12:07 TingTingin

+1 for this!

wwirving avatar Aug 08 '23 18:08 wwirving

Agree! Training Refiner model is really important now if community want to get a good anime model like NAI. They must train a anime refiner, only base fine-tune is not enough.

ymzlygw avatar Aug 09 '23 03:08 ymzlygw

+1 to this. SDXL base 1.0 is good, but the refiner is what elevates it to great.

singularity24601 avatar Sep 11 '23 09:09 singularity24601

+1 to this

yamkz avatar Oct 18 '23 04:10 yamkz

I want this too

FeiArt-Ai avatar Oct 24 '23 04:10 FeiArt-Ai

I want this too

MichaelFan01 avatar Nov 25 '23 08:11 MichaelFan01

Are there any docs or papers about the refiner? Something to start with?

mio-nyan avatar Nov 26 '23 14:11 mio-nyan

Seems like the scripts still do not support the refiner.

AbstractEyes avatar Mar 16 '25 21:03 AbstractEyes