fix(api): make sure ControlNet constructor is used for same pipeline (#449)
This commit is contained in:
parent
b7f2313489
commit
00db9a2787
|
@ -230,7 +230,7 @@ def load_pipeline(
|
|||
add_watermarker=False, # not so invisible: https://github.com/ssube/onnx-web/issues/438
|
||||
)
|
||||
else:
|
||||
if "controlnet" in components:
|
||||
if params.is_control():
|
||||
logger.debug(
|
||||
"assembling SD pipeline for %s with ControlNet",
|
||||
pipeline_class.__name__,
|
||||
|
|
|
@ -292,7 +292,7 @@ class ImageParams:
|
|||
return group
|
||||
|
||||
def is_control(self):
|
||||
return self.pipeline == "controlnet"
|
||||
return self.pipeline in ["controlnet", "controlnet-sdxl"]
|
||||
|
||||
def is_lpw(self):
|
||||
return self.pipeline == "lpw"
|
||||
|
|
Loading…
Reference in New Issue