wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-28 18:07:25 +08:00
Code Issues Actions 28 Packages Projects Releases Wiki Activity

whitespace

Browse Source
This commit is contained in:
kijai 2026-05-23 18:37:51 +03:00
parent 5ecaf09544
commit 2aba5bafca
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/lens/model.py
View File

@ -222,7 +222,7 @@ class LensJointAttention(nn.Module):
# [B, S, H, D] → [B, H, S, D] for attention, dels to avoid VRAM peaks # [B, S, H, D] → [B, H, S, D] for attention, dels to avoid VRAM peaks
q = torch.cat([img_q, txt_q], dim=1).transpose(1, 2) q = torch.cat([img_q, txt_q], dim=1).transpose(1, 2)
del img_q, txt_q del img_q, txt_q
k = torch.cat([img_k, txt_k], dim=1).transpose(1, 2) k = torch.cat([img_k, txt_k], dim=1).transpose(1, 2)
del img_k, txt_k del img_k, txt_k
v = torch.cat([img_v, txt_v], dim=1).transpose(1, 2) v = torch.cat([img_v, txt_v], dim=1).transpose(1, 2)