Linoy Tsaban b4077af212 [bug fix] using snr gamma and prior preservation loss in the dreambooth lora sdxl training scripts (#6356) ...

* change timesteps used to calculate snr when --with_prior_preservation is enabled

* change timesteps used to calculate snr when --with_prior_preservation is enabled (canonical script)

* style

* revert canonical script to before snr gamma change

---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

2024-01-02 09:21:39 -06:00

..

train_dreambooth_lora_sdxl_advanced.py

[bug fix] using snr gamma and prior preservation loss in the dreambooth lora sdxl training scripts (#6356)

2024-01-02 09:21:39 -06:00