examples/dreambooth/train_dreambooth.py · 7d631825b0bfcc2ee3f80cb3f39a8eec6dfa1bc5 · chenpangpang / diffusers

"src/vscode:/vscode.git/clone" did not exist on "5ee2f1729a02d1c041817beb8e5347bc56bc7112"

Make Dreambooth SD Training Script `torch.compile` compatible (#6532) · 7d631825

Vinh H. Pham authored Jan 12, 2024



* support compile

* make style

* move unwrap_model inside function

* change unwrap call

* run make style

* Update examples/dreambooth/train_dreambooth.py
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

* Revert "Update examples/dreambooth/train_dreambooth.py"

This reverts commit 70ab09732e7cfec0b19c497f823ddd1c8259dad0.

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

7d631825

train_dreambooth.py 56.4 KB

Replace train_dreambooth.py