fix(api): restore inpaint pipeline option

ssube · Apr 15, 2023 · b076326 · b076326
1 parent b20df68
commit b076326
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py
@@ -33,6 +33,8 @@
     KDPM2DiscreteScheduler,
     LMSDiscreteScheduler,
     OnnxRuntimeModel,
+    OnnxStableDiffusionImg2ImgPipeline,
+    OnnxStableDiffusionInpaintPipeline,
     OnnxStableDiffusionPipeline,
     PNDMScheduler,
     StableDiffusionPipeline,
@@ -46,11 +48,11 @@
 
 available_pipelines = {
     "controlnet": OnnxStableDiffusionControlNetPipeline,
-    # "img2img": OnnxStableDiffusionImg2ImgPipeline,
-    # "inpaint": OnnxStableDiffusionInpaintPipeline,
+    "img2img": OnnxStableDiffusionImg2ImgPipeline,
+    "inpaint": OnnxStableDiffusionInpaintPipeline,
     "lpw": OnnxStableDiffusionLongPromptWeightingPipeline,
     "pix2pix": OnnxStableDiffusionInstructPix2PixPipeline,
-    # "txt2img": OnnxStableDiffusionPipeline,
+    "txt2img": OnnxStableDiffusionPipeline,
 }
 
 pipeline_schedulers = {