src/diffusers/models/unet_2d.py · 9e234d8048e5b9f631937f6dcdec364952e4f90e · renzhc / diffusers_dcu

handle fp16 in `UNet2DModel` (#1216) · 9e234d80

Suraj Patil authored Nov 23, 2022



* make sure fp16 runs well

* add fp16 test for superes

* Update src/diffusers/models/unet_2d.py
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* gen on cuda

* always run fast inferecne test on cpu

* run on cpu
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

9e234d80

unet_2d.py 11.3 KB

Replace unet_2d.py