apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

add support for XL inpainting pipeline

This commit is contained in:
Sean Sube 2023-09-03 16:08:42 -05:00
parent 1d5ac7dde5
commit 9999a1bf56
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/diffusers/load.py
View File

@ -5,6 +5,7 @@ from typing import Any, List, Optional, Tuple
from onnx import load_model from onnx import load_model
from optimum.onnxruntime import ( from optimum.onnxruntime import (
ORTStableDiffusionXLImg2ImgPipeline, ORTStableDiffusionXLImg2ImgPipeline,
ORTStableDiffusionXLInpaintPipeline,
ORTStableDiffusionXLPipeline, ORTStableDiffusionXLPipeline,
) )
from optimum.onnxruntime.modeling_diffusion import ORTModelTextEncoder, ORTModelUnet from optimum.onnxruntime.modeling_diffusion import ORTModelTextEncoder, ORTModelUnet
@ -55,6 +56,7 @@ available_pipelines = {
"img2img": OnnxStableDiffusionImg2ImgPipeline, "img2img": OnnxStableDiffusionImg2ImgPipeline,
"img2img-sdxl": ORTStableDiffusionXLImg2ImgPipeline, "img2img-sdxl": ORTStableDiffusionXLImg2ImgPipeline,
"inpaint": OnnxStableDiffusionInpaintPipeline, "inpaint": OnnxStableDiffusionInpaintPipeline,
"inpaint-sdxl": ORTStableDiffusionXLInpaintPipeline,
"lpw": OnnxStableDiffusionLongPromptWeightingPipeline, "lpw": OnnxStableDiffusionLongPromptWeightingPipeline,
"panorama": OnnxStableDiffusionPanoramaPipeline, "panorama": OnnxStableDiffusionPanoramaPipeline,
"pix2pix": OnnxStableDiffusionInstructPix2PixPipeline, "pix2pix": OnnxStableDiffusionInstructPix2PixPipeline,