feat(api): add panorama pipeline for ONNX

ssube · Apr 26, 2023 · 214af7f · 214af7f
1 parent 42a4720
commit 214af7f
Show file tree

Hide file tree

Showing 2 changed files with 497 additions and 0 deletions.
diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py
@@ -17,6 +17,7 @@
 from ..utils import run_gc
 from .pipelines.controlnet import OnnxStableDiffusionControlNetPipeline
 from .pipelines.lpw import OnnxStableDiffusionLongPromptWeightingPipeline
+from .pipelines.panorama import OnnxStableDiffusionPanoramaPipeline
 from .pipelines.pix2pix import OnnxStableDiffusionInstructPix2PixPipeline
 from .version_safe_diffusers import (
     DDIMScheduler,
@@ -48,6 +49,7 @@
     "img2img": OnnxStableDiffusionImg2ImgPipeline,
     "inpaint": OnnxStableDiffusionInpaintPipeline,
     "lpw": OnnxStableDiffusionLongPromptWeightingPipeline,
+    "panorama": OnnxStableDiffusionPanoramaPipeline,
     "pix2pix": OnnxStableDiffusionInstructPix2PixPipeline,
     "txt2img": OnnxStableDiffusionPipeline,
     "upscale": OnnxStableDiffusionUpscalePipeline,