diffusers
dreambooth: fix #1566: maintain fp32 wrapper when saving a checkpoint to avoid crash when running fp16
#1618

Merged

dreambooth: fix #1566: maintain fp32 wrapper when saving a checkpoint to avoid crash when running fp16 #1618

patrickvonplaten merged 4 commits into huggingface:main from timh:fix-1566-dreambooth

dreambooth: fix #1566: maintain fp32 wrapper when saving a checkpoint…

80486c15

dreambooth: guard against passing keep_fp32_wrapper arg to older vers…

ebfd8bf2

timh force pushed from 053e6b7c to ebfd8bf2 3 years ago

pcuenca requested a review from

patil-suraj 3 years ago

pcuenca requested a review from

williamberman 3 years ago

pcuenca requested a review from

patrickvonplaten 3 years ago

pcuenca approved these changes on 2022-12-10

Apply suggestions from code review

7d039df2

patrickvonplaten commented on 2022-12-10

Update examples/dreambooth/train_dreambooth.py

94bd34d2

patrickvonplaten added training

patrickvonplaten merged 2868d991 into main 3 years ago

timh deleted the fix-1566-dreambooth branch 3 years ago

Reviewers

pcuenca

patrickvonplaten

patil-suraj

williamberman

Assignees

No one assigned

Labels

training

Milestone

No milestone