diffusers
Make Dreambooth SD Training Script `torch.compile` compatible
#6532

Merged

Make Dreambooth SD Training Script `torch.compile` compatible #6532

sayakpaul merged 7 commits into huggingface:main from rootonchair:compile_compatible_dreambooth

support compile

8ab999e5

make style

c4468f5f

sayakpaul commented on 2024-01-12

sayakpaul approved these changes on 2024-01-12

move unwrap_model inside function

40b21913

change unwrap call

4191fc30

run make style

362461ba

sayakpaul commented on 2024-01-12

sayakpaul approved these changes on 2024-01-12

Update examples/dreambooth/train_dreambooth.py

70ab0973

Revert "Update examples/dreambooth/train_dreambooth.py"

6e8b573d

sayakpaul approved these changes on 2024-01-12

sayakpaul merged 7d631825 into main 2 years ago

Reviewers

sayakpaul

Assignees

No one assigned

Labels

None yet

Milestone

No milestone