optimum
3bac3386 - Add vae image processor (#1219)

Commit

1 year ago

Add vae image processor (#1219) * add vae image processor * tests refactorization * make watermark optional * format * add shape test * add tests * format * raise error if watermark not installed but needed * fix image resizing for diffusers < v0.21.0

References

#1219 - add vae image processor

Author

echarlaix

Parents

7f8e6066

Files8

optimum
- onnxruntime
  - modeling_diffusion.py
- pipelines/diffusers
  - pipeline_stable_diffusion.py
  - pipeline_stable_diffusion_img2img.py
  - pipeline_stable_diffusion_inpaint.py
  - pipeline_stable_diffusion_xl.py
  - pipeline_stable_diffusion_xl_img2img.py
  - pipeline_utils.py
tests/onnxruntime
- test_stable_diffusion_pipeline.py

optimum 3bac3386 - Add vae image processor (#1219)

optimum
3bac3386 - Add vae image processor (#1219)