[diffusion] refactor: refactor attention backend checking to use backend enum (#15555)

Signed-off-by: Xiaodong Ye <yeahdongcn@gmail.com>
This commit is contained in:
R0CKSTAR
2025-12-25 11:51:40 +08:00
committed by GitHub
parent b311c43d13
commit 8865424f14
14 changed files with 53 additions and 93 deletions
@@ -23,6 +23,7 @@ logger = init_logger(__name__)
class AttentionBackendEnum(enum.Enum):
FA2 = enum.auto()
FA = enum.auto()
SLIDING_TILE_ATTN = enum.auto()
TORCH_SDPA = enum.auto()