examples/textual_inversion/textual_inversion_flax.py · 0df83c79e4247e6b58c4c0aacfcb40b74db8d96e · 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

textual_inversion_flax.py 24.2 KB

Replace textual_inversion_flax.py