diffusion: support fa4 in fa backend for blackwell (#13263)

Co-authored-by: Mick <mickjagger19@icloud.com>
2025-11-16 21:02:45 +08:00
parent 191f5c7795
commit 6afe396399
15 changed files with 55 additions and 34 deletions
@@ -86,7 +86,7 @@ class CausalWanSelfAttention(nn.Module):
            softmax_scale=None,
            causal=False,
            supported_attention_backends=(
-                AttentionBackendEnum.FA3,
+                AttentionBackendEnum.FA,
                AttentionBackendEnum.TORCH_SDPA,
            ),
        )