1
0
Fork 0

fix(api): add loading path for controlnet

This commit is contained in:
Sean Sube 2023-12-18 22:21:33 -06:00
parent 6708efa8ce
commit a716f6db1f
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 17 additions and 1 deletions

View File

@ -229,7 +229,23 @@ def load_pipeline(
tokenizer_2=components.get("tokenizer_2", None),
)
else:
if "vae" in components:
if "controlnet" in components:
logger.debug(
"assembling SD pipeline for %s with ControlNet",
pipeline_class.__name__,
)
pipe = pipeline_class(
components["vae"],
components["text_encoder"],
components["tokenizer"],
components["unet"],
components["controlnet"],
scheduler,
None,
None,
requires_safety_checker=False,
)
elif "vae" in components:
# upscale uses a single VAE
logger.debug(
"assembling SD pipeline for %s with single VAE",