PR #1216 handle fp16 in `UNet2DModel`

handle fp16 in `UNet2DModel` #1216

patil-suraj merged 7 commits into main from unet_2d_fp16

make sure fp16 runs well

5e2ac9fa

patil-suraj requested a review from

patrickvonplaten 3 years ago

patrickvonplaten commented on 2022-11-09

patrickvonplaten assigned

patil-suraj 2 years ago

pcuenca approved these changes on 2022-11-22

add fp16 test for superes

3eee4701

Update src/diffusers/models/unet_2d.py

5f7c48bc

gen on cuda

65ff1b3f

always run fast inferecne test on cpu

1080fdaf

run on cpu

5d203285

Merge branch 'main' into unet_2d_fp16

002c07d6

patil-suraj commented on 2022-11-23

patil-suraj requested a review from

patrickvonplaten 2 years ago

patil-suraj merged 9e234d80 into main 2 years ago

patil-suraj deleted the unet_2d_fp16 branch 2 years ago

Reviewers

pcuenca

patrickvonplaten

Assignees

patil-suraj

Labels

None yet

Milestone

No milestone