From ea0f939df32cc808ac1ce79448d56eeef89796e2 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Tue, 25 Feb 2025 19:13:39 -0500 Subject: [PATCH] Fix issue with wan and other attention implementations. --- comfy/ldm/wan/model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/comfy/ldm/wan/model.py b/comfy/ldm/wan/model.py index 546ebb2253a..151192975fc 100644 --- a/comfy/ldm/wan/model.py +++ b/comfy/ldm/wan/model.py @@ -88,9 +88,9 @@ def qkv_fn(x): q, k = apply_rope(q, k, freqs) x = optimized_attention( - q=q.view(b, s, n * d), - k=k.view(b, s, n * d), - v=v, + q.view(b, s, n * d), + k.view(b, s, n * d), + v, heads=self.num_heads, )