Remove now unnecessary minimax memory workaround. (#16103)

This commit is contained in:
comfyanonymous
2026-09-04 10:27:35 -07:00
committed by GitHub
parent 804eb5513a
commit 5c23fb7b6c
+1 -1
View File
@@ -188,7 +188,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))