xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-16 00:16:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix Flux Kontext ReferenceLatent and ControlNet shape mismatch

Browse Source
This commit is contained in:
Ivan R. 2025-08-05 04:48:13 +05:00
parent 03895dea7c
commit 9ce912099e
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/ldm/flux/model.py
View File

@ -157,6 +157,11 @@ class Flux(nn.Module):
if i < len(control_i): if i < len(control_i):
add = control_i[i] add = control_i[i]
if add is not None: if add is not None:
if img.shape[1] != add.shape[1]:
padding_size = img.shape[1] - add.shape[1]
if padding_size > 0:
padding = torch.zeros(add.shape[0], padding_size, add.shape[2], device=add.device, dtype=add.dtype)
add = torch.cat([add, padding], dim=1)
img += add img += add
if img.dtype == torch.float16: if img.dtype == torch.float16: