search

kohya-ss / sd-scripts

Apache License 2.0
5.33k stars 881 forks source link

Request: Add Support for V-Prediction SDXL Model in LoRA Training #1772

Closed Wenaka2004 closed 2 weeks ago

Wenaka2004 commented 3 weeks ago

Please add support for training LoRA models with v-prediction SDXL models. The v-prediction version of the NoobAI_XL model has shown excellent performance, and adding this feature would enable further training based on it.

Currently, training LoRA on v-prediction models does not throw errors but results in empty images during generation.

kohya-ss commented 3 weeks ago

The dev branch will work with v-prediction SDXL model. NoobAI_XL may assume zero terminal snr. If so, please add --zero_terminal_snr option.

Wenaka2004 commented 2 weeks ago

Thank you for your response and the solution! I’ll give it a try.

Wenaka2004 commented 2 weeks ago

Training has been successful, thanks.

nephi-dev commented 1 week ago

does this work on the schedule free branch?