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>