zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-14 21:54:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix peak memory issue with H3. (#15486)

Browse Source
This commit is contained in:
comfyanonymous
2026-08-11 02:09:29 -07:00
committed by GitHub
parent bf4c9a08fc
commit 62b3c94bd4
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/ldm/minimax/model.py
View File

@@ -165,6 +165,7 @@ class Attention(nn.Module):
else:
q = self.q_norm(q.view(s, self.heads, self.head_dim))
k = self.k_norm(k.view(s, self.heads, self.head_dim))
v = v.clone()
q = AttentionTensorContainer(q.transpose(0, 1).unsqueeze(0))
k = AttentionTensorContainer(k.transpose(0, 1).unsqueeze(0))
v = AttentionTensorContainer(v.transpose(0, 1).unsqueeze(0))