From 93b02261b2b594cda63dd8ccd907045f2ec80382 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 29 Apr 2023 18:50:26 -0500 Subject: [PATCH] fix(api): use img2img pipeline for loopback when controlnet was originally selected --- api/onnx_web/diffusers/run.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py index 72fc85ee..f869c409 100644 --- 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(