From 682a182ce5f53850afb15672bf7395ee2f78b102 Mon Sep 17 00:00:00 2001 From: STwangyingrui Date: Wed, 26 Aug 2026 10:45:17 +0000 Subject: [PATCH] fix: enable SageAttention dispatcher on SM120 --- lightx2v/common/ops/attn/sage_attn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lightx2v/common/ops/attn/sage_attn.py b/lightx2v/common/ops/attn/sage_attn.py index 085adc1a6..07945b9f5 100755 --- a/lightx2v/common/ops/attn/sage_attn.py +++ b/lightx2v/common/ops/attn/sage_attn.py @@ -25,7 +25,7 @@ capability = torch.cuda.get_device_capability(0) if torch.cuda.is_available() else None # Keep the legacy SM89 override; SM120 follows SageAttention's dispatcher. -if capability in [(8, 9), (12, 0)]: +if capability == (8, 9): try: from sageattention import sageattn_qk_int8_pv_fp16_triton as sageattn except ImportError: