From 14208de393dcf7d1302a5eed70679f246f63c57e Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 23 Dec 2023 22:18:09 -0600
Subject: [PATCH] register optimum-based SD converter

---
 api/onnx_web/convert/diffusion/diffusion.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/api/onnx_web/convert/diffusion/diffusion.py b/api/onnx_web/convert/diffusion/diffusion.py
index 6d2652bd..71de8003 100644
--- a/api/onnx_web/convert/diffusion/diffusion.py
+++ b/api/onnx_web/convert/diffusion/diffusion.py
@@ -59,6 +59,7 @@ CONVERT_PIPELINES = {
     "panorama": StableDiffusionPipeline,
     "pix2pix": StableDiffusionInstructPix2PixPipeline,
     "txt2img": StableDiffusionPipeline,
+    "txt2img-optimum": StableDiffusionPipeline,
     "upscale": StableDiffusionUpscalePipeline,
 }
 
@@ -781,7 +782,10 @@ def convert_diffusion_diffusers_optimum(
 
     # diffusers go into a directory rather than .onnx file
     logger.info(
-        "converting Stable Diffusion model %s: %s -> %s/", name, source, dest_path
+        "converting Stable Diffusion model %s using Optimum: %s -> %s/",
+        name,
+        source,
+        dest_path,
     )
 
     if path.exists(dest_path) and path.exists(model_index):