fix pulid

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-01-10 08:00:17 -05:00
parent 6030e95016
commit 465d7caa3a
4 changed files with 9 additions and 3 deletions
+3 -1
View File
@@ -410,7 +410,9 @@ def control_run(state: str = '',
blended_image = None
# set pipeline
if pipe.__class__.__name__ != shared.sd_model.__class__.__name__:
if pipe is None:
return [], '', '', 'Pipeline not set'
elif pipe.__class__.__name__ != shared.sd_model.__class__.__name__:
original_pipeline = shared.sd_model
shared.sd_model = pipe
sd_models.move_model(shared.sd_model, shared.device)