wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-27 17:37:39 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Lens: remove redundant memory optimisation

Browse Source
This commit is contained in:
kijai 2026-05-26 09:27:44 +03:00
parent d16fea7105
commit 893915b949
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -149,8 +149,6 @@ class LensJointAttention(nn.Module):
txt_q, txt_k, txt_v = txt_qkv.unbind(dim=2)
txt_q = self.norm_added_q(txt_q)
txt_k = self.norm_added_k(txt_k)
txt_v = txt_v.contiguous()
del txt_qkv
# [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)