apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): use img2img pipeline for loopback when controlnet was originally selected

This commit is contained in:
Sean Sube 2023-04-29 18:50:26 -05:00
parent 47c7f5d1da
commit 93b02261b2
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/diffusers/run.py
View File

@ -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(