apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): make sure ControlNet constructor is used for same pipeline (#449)

This commit is contained in:
Sean Sube 2023-12-31 13:00:34 -06:00
parent b7f2313489
commit 00db9a2787
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/diffusers/load.py
View File

@ -230,7 +230,7 @@ def load_pipeline(
add_watermarker=False, # not so invisible: https://github.com/ssube/onnx-web/issues/438 add_watermarker=False, # not so invisible: https://github.com/ssube/onnx-web/issues/438
) )
else: else:
if "controlnet" in components: if params.is_control():
logger.debug( logger.debug(
"assembling SD pipeline for %s with ControlNet", "assembling SD pipeline for %s with ControlNet",
pipeline_class.__name__, pipeline_class.__name__,

2
api/onnx_web/params.py
View File

@ -292,7 +292,7 @@ class ImageParams:
return group return group
def is_control(self): def is_control(self):
return self.pipeline == "controlnet" return self.pipeline in ["controlnet", "controlnet-sdxl"]
def is_lpw(self): def is_lpw(self):
return self.pipeline == "lpw" return self.pipeline == "lpw"