1
0
Fork 0

fix(api): make sure a controlnet has been selected when using that pipeline

This commit is contained in:
Sean Sube 2024-01-12 22:28:59 -06:00
parent f30c5f2d31
commit 7966f001e1
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 0 deletions

View File

@ -231,6 +231,9 @@ def load_pipeline(
)
else:
if params.is_control():
if "controlnet" not in components or components["controlnet"] is None:
raise ValueError("ControlNet is required for control pipelines")
logger.debug(
"assembling SD pipeline for %s with ControlNet",
pipeline_class.__name__,