diffusers
fix vae dtype when accelerate config using --mixed_precision="fp16"
#9601
Merged

fix vae dtype when accelerate config using --mixed_precision="fp16" #9601

xduzhangjiayu
xduzhangjiayu fix vae dtype when accelerate config using --mixed_precision="fp16"
e7ef7b60
sayakpaul
sayakpaul commented on 2024-10-07
xduzhangjiayu Add param for upcast vae
be4c8102
xduzhangjiayu
HuggingFaceDocBuilderDev
sayakpaul
sayakpaul approved these changes on 2024-10-07
sayakpaul sayakpaul merged 2cb383f5 into main 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone