examples/dreambooth/train_dreambooth_flax.py · f106ab40b316f4e98b06e2070c4cd34a976f9ee6 · OpenDAS / diffusers

[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

train_dreambooth_flax.py 24.8 KB

Replace train_dreambooth_flax.py