From 2d60357fb8bcd928d65f7435b70c7ad178225360 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=8C=80=E7=80=9A?= Date: Sat, 28 Mar 2026 02:10:12 +0800 Subject: [PATCH] Fix the parameter name error. fa_version -> version which is defined in the wan/modules/attention.py:attention --- Matrix-Game-3/wan/distributed/ulysses.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Matrix-Game-3/wan/distributed/ulysses.py b/Matrix-Game-3/wan/distributed/ulysses.py index e9cd669..c7ee3b4 100644 --- a/Matrix-Game-3/wan/distributed/ulysses.py +++ b/Matrix-Game-3/wan/distributed/ulysses.py @@ -66,7 +66,7 @@ def distributed_attention( v, k_lens=seq_lens, window_size=window_size, - fa_version=fa_version, + version=fa_version, ) # scatter q/k/v sequence