diffusers
b2c9b546 - [img2img, inpainting] fix fp16 inference (#769)

Commit

2 years ago

[img2img, inpainting] fix fp16 inference (#769) * handle dtype in vae and image2image pipeline * fix inpaint in fp16 * dtype should be handled in add_noise * style * address review comments * add simple fast tests to check fp16 * fix test name * put mask in fp16

References

patch_release_v0_4_2

Author

patil-suraj

Committer

patrickvonplaten

Parents

7a6cf891

Files4

src/diffusers
- models
  - vae.py
- pipelines/stable_diffusion
  - pipeline_stable_diffusion_img2img.py
  - pipeline_stable_diffusion_inpaint.py
tests
- test_pipelines.py

diffusers b2c9b546 - [img2img, inpainting] fix fp16 inference (#769)

diffusers
b2c9b546 - [img2img, inpainting] fix fp16 inference (#769)