keep_fp32_wrapper

[ZeroCool22]

Describe the bug

Reproduction

Attempt to save at step 500 and get the error.

Logs

No response

System Info

• diffusers version: 0.9.0
• Platform: Linux-5.10.102.1-microsoft-standard-WSL2-x86_64-with-glibc2.31
• Python version: 3.9.13
• PyTorch version (GPU?): 1.12.1+cu116 (True)
• Huggingface_hub version: 0.10.0
• Transformers version: 4.24.0
• Using GPU in script?:
• Using distributed or parallel set-up in script?:

[ShivamShrirao]

Update your accelerate version.