Linoy Tsaban 3adc804f79 [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-12-23 13:02:05 +05:30

..

train_dreambooth_lora_sdxl_advanced.py

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

2024-12-23 13:02:05 +05:30