tests/test_scheduler_flax.py · f106ab40b316f4e98b06e2070c4cd34a976f9ee6 · renzhc / diffusers_dcu

[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

test_scheduler_flax.py 40.9 KB

Replace test_scheduler_flax.py