examples/dreambooth/train_dreambooth_lora_sd3.py · c4b5d2ff6b529ac0f895cedb04fef5b25e89c412 · renzhc / diffusers_dcu

[SD3 dreambooth lora] smol fix to checkpoint saving (#9993) · c4b5d2ff

Linoy Tsaban authored Nov 24, 2024

* smol change to fix checkpoint saving & resuming (as done in train_dreambooth_sd3.py)

* style

* modify comment to explain reasoning behind hidden size check

c4b5d2ff

train_dreambooth_lora_sd3.py 78.7 KB

Replace train_dreambooth_lora_sd3.py