fix(api): use img2img pipeline for loopback when controlnet was origi…

…nally selected
ssube · Apr 29, 2023 · 93b0226 · 93b0226
1 parent 47c7f5d
commit 93b0226
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py
@@ -48,6 +48,10 @@ def run_loopback(
 
     # load img2img pipeline once
     pipe_type = params.get_valid_pipeline("img2img")
+    if pipe_type == "controlnet":
+        logger.debug("controlnet pipeline cannot be used for loopback, switching to img2img")
+        pipe_type = "img2img"
+
     logger.debug("using %s pipeline for loopback", pipe_type)
 
     pipe = pipeline or load_pipeline(