From ef781cabcb9e6ee7ea24052d3b10008cad9c2024 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Wed, 7 Feb 2024 13:02:42 -0800 Subject: [PATCH] Backend: Allow control signal to be none for advanced weighting --- ldm_patched/modules/controlnet.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/ldm_patched/modules/controlnet.py b/ldm_patched/modules/controlnet.py index 3cd4d141..e25a4779 100644 --- a/ldm_patched/modules/controlnet.py +++ b/ldm_patched/modules/controlnet.py @@ -64,6 +64,10 @@ def compute_controlnet_weighting(control, cnet): for k, v in control.items(): for i in range(len(v)): control_signal = control[k][i] + + if not isinstance(control_signal, torch.Tensor): + continue + B, C, H, W = control_signal.shape positive_weight = 1.0