examples/controlnet/train_controlnet_flax.py · 8de78001df95a641bf6ef942bee9553921d44490 · chenpangpang / diffusers

"docs/source/en/training/text_inversion.md" did not exist on "78744b6a8f3c9dd4800e1b279cc37930dfd77048"

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