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

Merge 40e7284ab3ed07c816ac03b3d50218851514153e into b20ba1f27cbd4e1c84cf8ec72b345723de9e7c80

Browse Source
This commit is contained in:
Ivan R. 2025-08-28 01:05:17 +07:00 committed by GitHub
commit 1bd59a7e2f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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

@ -158,6 +158,11 @@ class Flux(nn.Module):
if i < len(control_i):
add = control_i[i]
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
if img.dtype == torch.float16: