Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

multi controlnet-union

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-02-13 07:54:47 -05:00
parent 5dacd918af
commit 9ce34b4c8c
4 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/control/units/controlnet.py
+4 -7
View File
@@ -338,14 +338,11 @@ class ControlNetPipeline():
classes = [c.__class__.__name__ for c in controlnets]
if any(c == 'ControlNetUnionModel' for c in classes):
if not all(c == 'ControlNetUnionModel' for c in classes):
log.warning(f'Control {what}: units={classes} mixed type')
log.warning(f'Control {what}: units={classes} mixed type is not supported')
return
if isinstance(controlnets, list) and len(controlnets) == 1:
controlnets = controlnets[0]
cls = StableDiffusionXLControlNetUnionPipeline
if len(controlnets) > 1:
# TODO controlnet-union multi-unit
controlnets = controlnets[0]
log.warning(f'Control {what}: units={classes} supports single unit only')
else:
controlnets = controlnets[0]
else:
cls = StableDiffusionXLControlNetPipeline
self.pipeline = cls(