[diffusion][Wan]: fix sparse attention backends being applied to cross-attention (#17596)

This commit is contained in:
Ratish P
2026-02-16 21:57:58 +08:00
committed by GitHub
parent b86c6491fa
commit c1d1337afc
3 changed files with 28 additions and 6 deletions
@@ -41,6 +41,16 @@ class AttentionBackendEnum(enum.Enum):
def __str__(self):
return self.name.lower()
@property
def is_sparse(self) -> bool:
return self in {
AttentionBackendEnum.SLIDING_TILE_ATTN,
AttentionBackendEnum.VIDEO_SPARSE_ATTN,
AttentionBackendEnum.VMOBA_ATTN,
AttentionBackendEnum.SLA_ATTN,
AttentionBackendEnum.SAGE_SLA_ATTN,
}
class PlatformEnum(enum.Enum):
CUDA = enum.auto()