src/diffusers/schedulers/scheduling_karras_ve_flax.py · f106ab40b316f4e98b06e2070c4cd34a976f9ee6 · renzhc / diffusers_dcu

"tests/models/test_models_unet_2d_condition.py" did not exist on "f1484b81b0b763aa010fd2083df933305afb1812"

[Flax] Stateless schedulers, fixes and refactors (#1661) · f106ab40

Simon Kirsten authored Dec 20, 2022



* [Flax] Stateless schedulers, fixes and refactors

* Remove scheduling_common_flax and some renames

* Update src/diffusers/schedulers/scheduling_pndm_flax.py
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

f106ab40

scheduling_karras_ve_flax.py 9.41 KB

Replace scheduling_karras_ve_flax.py