Fix peak memory issue with H3. (#15486)

2026-08-11 02:09:29 -07:00 · 2026-08-11 02:09:29 -07:00 · 62b3c94bd4
parent bf4c9a08fc
commit 62b3c94bd4
1 changed files with 1 additions and 0 deletions
--- a/comfy/ldm/minimax/model.py
+++ b/comfy/ldm/minimax/model.py
@ -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))