examples/controlnet/train_controlnet_flax.py · e6faf607f71b86bf210cc4eca06555304f445611 · renzhc / diffusers_dcu

Min-SNR Gamma: follow-up fix for zero-terminal SNR models on v-prediction or epsilon (#5177) · 4a06c745

Bagheera authored Sep 26, 2023



* merge with main

* fix flax example

* fix onnx example

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

4a06c745

train_controlnet_flax.py 43.9 KB

Replace train_controlnet_flax.py