examples/text_to_image/train_text_to_image.py · 8de78001df95a641bf6ef942bee9553921d44490 · OpenDAS / diffusers

"test/spectests.cpp" did not exist on "a725d4b190e8cbd1ac203d09746dc8e65bece21d"

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_text_to_image.py 43.6 KB

Replace train_text_to_image.py