tests/pipelines/latent_diffusion/test_latent_diffusion_superresolution.py · 9e234d8048e5b9f631937f6dcdec364952e4f90e · chenpangpang / diffusers

"vscode:/vscode.git/clone" did not exist on "4fa3fcbb3b6d1dcef2a153e086914a5052555645"

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

test_latent_diffusion_superresolution.py 4.69 KB

Replace test_latent_diffusion_superresolution.py