examples/controlnet/train_controlnet_flax.py · 8de78001df95a641bf6ef942bee9553921d44490 · renzhc / diffusers_dcu

Correct SNR weighted loss in v-prediction case by only adding 1 to SNR on the denominator (#6307) · c1010662

Yingtian Liu authored Jan 27, 2024



* fix minsnr implementation for v-prediction case

* format code

* always compute snr when snr_gamma is specified

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

c1010662

train_controlnet_flax.py 43.5 KB

Replace train_controlnet_flax.py