From 9999a1bf561727c1b1042af99d6f13cdd0fe33db Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sun, 3 Sep 2023 16:08:42 -0500
Subject: [PATCH] add support for XL inpainting pipeline

---
 api/onnx_web/diffusers/load.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py
index 9b9c9d78..95f47fd2 100644
--- a/api/onnx_web/diffusers/load.py
+++ b/api/onnx_web/diffusers/load.py
@@ -5,6 +5,7 @@ from typing import Any, List, Optional, Tuple
 from onnx import load_model
 from optimum.onnxruntime import (
     ORTStableDiffusionXLImg2ImgPipeline,
+    ORTStableDiffusionXLInpaintPipeline,
     ORTStableDiffusionXLPipeline,
 )
 from optimum.onnxruntime.modeling_diffusion import ORTModelTextEncoder, ORTModelUnet
@@ -55,6 +56,7 @@ available_pipelines = {
     "img2img": OnnxStableDiffusionImg2ImgPipeline,
     "img2img-sdxl": ORTStableDiffusionXLImg2ImgPipeline,
     "inpaint": OnnxStableDiffusionInpaintPipeline,
+    "inpaint-sdxl": ORTStableDiffusionXLInpaintPipeline,
     "lpw": OnnxStableDiffusionLongPromptWeightingPipeline,
     "panorama": OnnxStableDiffusionPanoramaPipeline,
     "pix2pix": OnnxStableDiffusionInstructPix2PixPipeline,