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
@@ -20,7 +20,7 @@ class DiTArchConfig(ArchConfig):
        default_factory=lambda: {
            AttentionBackendEnum.SLIDING_TILE_ATTN,
            AttentionBackendEnum.SAGE_ATTN,
-            AttentionBackendEnum.FA3,
+            AttentionBackendEnum.FA,
            AttentionBackendEnum.TORCH_SDPA,
            AttentionBackendEnum.VIDEO_SPARSE_ATTN,
            AttentionBackendEnum.VMOBA_ATTN,
@@ -16,7 +16,7 @@ class EncoderArchConfig(ArchConfig):
    architectures: list[str] = field(default_factory=lambda: [])
    _supported_attention_backends: set[AttentionBackendEnum] = field(
        default_factory=lambda: {
-            AttentionBackendEnum.FA3,
+            AttentionBackendEnum.FA,
            AttentionBackendEnum.TORCH_SDPA,
        }
    )