diffusers
2cb383f5 - fix vae dtype when accelerate config using --mixed_precision="fp16" (#9601)

Commit

1 year ago

fix vae dtype when accelerate config using --mixed_precision="fp16" (#9601) * fix vae dtype when accelerate config using --mixed_precision="fp16" * Add param for upcast vae

References

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

Author

xduzhangjiayu

Parents

31010ecc

diffusers 2cb383f5 - fix vae dtype when accelerate config using --mixed_precision="fp16" (#9601)

diffusers
2cb383f5 - fix vae dtype when accelerate config using --mixed_precision="fp16" (#9601)