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

[Dreambooth] flax fixes (#1765) · 9f8c915a

Pedro Cuenca authored Dec 19, 2022

* Fail if there are less images than the effective batch size.

* Remove lr-scheduler arg as it's currently ignored.

* Make guidance_scale work for batch_size > 1.

9f8c915a

train_dreambooth_flax.py 24.8 KB

Replace train_dreambooth_flax.py