Skip to content

Commit 3022090

Browse files
ctsimsanton-l
andauthored
Legacy Inpainting Pipeline for Onnx Models (#1237)
* Add legacy inpainting pipeline compatibility for onnx * remove commented out line * Add onnx legacy inpainting test * Fix slow decorators * pep8 styling * isort styling * dummy object * ordering consistency * style * docstring styles * Refactor common prompt encoding pattern * Update tests to permanent repository home * support all available schedulers until ONNX IO binding is available Co-authored-by: Anton Lozhkov <[email protected]> * updated styling from PR suggested feedback Co-authored-by: Anton Lozhkov <[email protected]>
1 parent 7240318 commit 3022090

File tree

6 files changed

+560
-0
lines changed

6 files changed

+560
-0
lines changed

src/diffusers/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@
8282
from .pipelines import (
8383
OnnxStableDiffusionImg2ImgPipeline,
8484
OnnxStableDiffusionInpaintPipeline,
85+
OnnxStableDiffusionInpaintPipelineLegacy,
8586
OnnxStableDiffusionPipeline,
8687
StableDiffusionOnnxPipeline,
8788
)

src/diffusers/pipelines/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
from .stable_diffusion import (
3131
OnnxStableDiffusionImg2ImgPipeline,
3232
OnnxStableDiffusionInpaintPipeline,
33+
OnnxStableDiffusionInpaintPipelineLegacy,
3334
OnnxStableDiffusionPipeline,
3435
StableDiffusionOnnxPipeline,
3536
)

src/diffusers/pipelines/stable_diffusion/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ class StableDiffusionPipelineOutput(BaseOutput):
3939
from .pipeline_onnx_stable_diffusion import OnnxStableDiffusionPipeline, StableDiffusionOnnxPipeline
4040
from .pipeline_onnx_stable_diffusion_img2img import OnnxStableDiffusionImg2ImgPipeline
4141
from .pipeline_onnx_stable_diffusion_inpaint import OnnxStableDiffusionInpaintPipeline
42+
from .pipeline_onnx_stable_diffusion_inpaint_legacy import OnnxStableDiffusionInpaintPipelineLegacy
4243

4344
if is_transformers_available() and is_flax_available():
4445
import flax

0 commit comments

Comments
 (0)