src/diffusers/models/unet_2d_condition.py · 051b34635fda2fc310898a6a602c89be8663b77f · renzhc / diffusers_dcu

[Half precision] Make sure half-precision is correct (#182) · 051b3463

Patrick von Platen authored Aug 16, 2022



* [Half precision] Make sure half-precision is correct

* Update src/diffusers/models/unet_2d.py

* Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py

* correct some tests

* Apply suggestions from code review
Co-authored-by: Suraj Patil <surajp815@gmail.com>

* finalize

* finish
Co-authored-by: Suraj Patil <surajp815@gmail.com>

051b3463

unet_2d_condition.py 6.84 KB

Replace unet_2d_condition.py