examples/dreambooth/train_dreambooth_lora_flux.py · acf479bded1ae9fc77b2dc7f316be099abf379ce · renzhc / diffusers_dcu

[advanced flux training] bug fix + reduce memory cost as in #9829 (#9838) · acf479bd

Linoy Tsaban authored Nov 18, 2024

* memory improvement as done here: https://github.com/huggingface/diffusers/pull/9829



* fix bug

* fix bug

* style

---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

acf479bd

train_dreambooth_lora_flux.py 75.6 KB

Replace train_dreambooth_lora_flux.py