examples/dreambooth/train_dreambooth_flax.py · 9f8c915a756cf937a428186a2af8cc56f029f689 · 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