[diffusion][Wan]: fix sparse attention backends being applied to cross-attention (#17596)
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user