tests/pipelines/stable_diffusion/test_onnx_stable_diffusion.py · 7b2407f4d75aaff406caf67808676d205c58d389 · OpenDAS / diffusers

add support for pre-calculated prompt embeds to Stable Diffusion ONNX pipelines (#2597) · 7b2407f4

Sean Sube authored Apr 12, 2023

* add support for prompt embeds to SD ONNX pipeline

* fix up the pipeline copies

* add prompt embeds param to other ONNX pipelines

* fix up prompt embeds param for SD upscaling ONNX pipeline

* add missing type annotations to ONNX pipes

7b2407f4

test_onnx_stable_diffusion.py 14.1 KB

Replace test_onnx_stable_diffusion.py