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

"vscode:/vscode.git/clone" did not exist on "4759d879f2376ffb9b82f296e442ec8ef137f27b"

[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_lms_discrete_flax.py 10.8 KB

Replace scheduling_lms_discrete_flax.py