apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

remove SDXL inpaint import

This commit is contained in:
Sean Sube 2023-09-05 17:05:27 -05:00
parent 1f1cc14efe
commit edfd059b46
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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