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

Commit
2 years ago
Make Dreambooth SD Training Script `torch.compile` compatible (#6532) * 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>
Author
Parents
Loading