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

Merged

Commits

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

timh committed 3 years ago
dreambooth: guard against passing keep_fp32_wrapper arg to older versions of accelerate. part of fix for #1566

timh committed 3 years ago
Apply suggestions from code review

patrickvonplaten committed 3 years ago
Update examples/dreambooth/train_dreambooth.py

patrickvonplaten committed 3 years ago