fix(api): add loading path for controlnet
This commit is contained in:
parent
6708efa8ce
commit
a716f6db1f
|
@ -229,7 +229,23 @@ def load_pipeline(
|
||||||
tokenizer_2=components.get("tokenizer_2", None),
|
tokenizer_2=components.get("tokenizer_2", None),
|
||||||
)
|
)
|
||||||
else:
|
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
|
# upscale uses a single VAE
|
||||||
logger.debug(
|
logger.debug(
|
||||||
"assembling SD pipeline for %s with single VAE",
|
"assembling SD pipeline for %s with single VAE",
|
||||||
|
|
Loading…
Reference in New Issue