lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-26 09:18:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Backend: Allow control signal to be none for advanced weighting

Browse Source
This commit is contained in:
lllyasviel
2024-02-07 13:02:42 -08:00
parent c3a66b016b
commit ef781cabcb
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ldm_patched/modules/controlnet.py
View File

@@ -64,6 +64,10 @@ def compute_controlnet_weighting(control, cnet):
for k, v in control.items(): for k, v in control.items():
for i in range(len(v)): for i in range(len(v)):
control_signal = control[k][i] control_signal = control[k][i]
if not isinstance(control_signal, torch.Tensor):
continue
B, C, H, W = control_signal.shape B, C, H, W = control_signal.shape
positive_weight = 1.0 positive_weight = 1.0