lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-07 14:09:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Backend: better controlnet mask batch broadcasting

Browse Source
This commit is contained in:
lllyasviel
2024-02-06 20:13:09 -08:00
parent 1110183943
commit c185e39e59
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ldm_patched/modules/controlnet.py
View File

@@ -87,8 +87,10 @@ def compute_controlnet_weighting(control, cnet):
final_weight = final_weight * sigma_weight * frame_weight
if isinstance(advanced_mask_weighting, torch.Tensor):
if control_signal.shape[0] == 2 * advanced_mask_weighting.shape[0]:
advanced_mask_weighting = advanced_mask_weighting.repeat(2, 1, 1, 1)
if advanced_mask_weighting.shape[0] != 1:
k = int(control_signal.shape[0] // advanced_mask_weighting.shape[0])
if control_signal.shape[0] == k * advanced_mask_weighting.shape[0]:
advanced_mask_weighting = advanced_mask_weighting.repeat(k, 1, 1, 1)
control_signal = control_signal * torch.nn.functional.interpolate(advanced_mask_weighting.to(control_signal), size=(H, W), mode='bilinear')
control[k][i] = control_signal * final_weight[:, None, None, None]