From 62b3c94bd45154f6486c7abf1b9efcacee96ea69 Mon Sep 17 00:00:00 2001 From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com> Date: Tue, 11 Aug 2026 02:09:29 -0700 Subject: [PATCH] Fix peak memory issue with H3. (#15486) --- comfy/ldm/minimax/model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/comfy/ldm/minimax/model.py b/comfy/ldm/minimax/model.py index 76174483a..f745db884 100644 --- 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))