From 5c018a9f376efc50cdbf92cf24ee13517e2c23ba Mon Sep 17 00:00:00 2001
From: Wenjie Dong <128051718+Explorer-Dong@users.noreply.github.com>
Date: Thu, 8 Jan 2026 13:47:22 +0800
Subject: [PATCH] fix: flash-attn-3 error

---
 wan/modules/attention.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wan/modules/attention.py b/wan/modules/attention.py
index 4dbbe03f..c8f69c2e 100644
--- a/wan/modules/attention.py
+++ b/wan/modules/attention.py
@@ -107,7 +107,7 @@ def half(x):
             max_seqlen_k=lk,
             softmax_scale=softmax_scale,
             causal=causal,
-            deterministic=deterministic)[0].unflatten(0, (b, lq))
+            deterministic=deterministic).unflatten(0, (b, lq))
     else:
         assert FLASH_ATTN_2_AVAILABLE
         x = flash_attn.flash_attn_varlen_func(