src/diffusers/schedulers/scheduling_pndm_flax.py · 3dacbb94ca25b7db876778aefbc41f9984b919e5 · renzhc / diffusers_dcu

`trained_betas` ignored in some schedulers (#635) · 3dacbb94

V Vishnu Anirudh authored Sep 29, 2022

* correcting the beta value assignment

* updating DDIM and LMSDiscreteFlax schedulers

* bringing back the changes that were lost as part of main branch merge

3dacbb94

scheduling_pndm_flax.py 21.2 KB

Replace scheduling_pndm_flax.py