diff --git a/pyproject.toml b/pyproject.toml
index 182726ae9..019d1a71c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -245,8 +245,8 @@ torchaudio = [
     { index = "pytorch-cpu", extra = "cpu" },
 ]
 sageattention = [
-    { git = "https://github.com/woct0rdho/SageAttention.git", rev = "v2.2.0", marker = "sys_platform == 'Linux'" },
-    { url = "https://github.com/woct0rdho/SageAttention/releases/download/v2.2.0-windows.post1/sageattention-2.2.0+cu128torch2.7.1.post1-cp39-abi3-win_amd64.whl", marker = "sys_platform == 'win32'" },
+    { git = "https://github.com/thu-ml/SageAttention.git", marker = "sys_platform == 'Linux'", extra = "attention" },
+    { url = "https://github.com/woct0rdho/SageAttention/releases/download/v2.2.0-windows.post1/sageattention-2.2.0+cu128torch2.7.1.post1-cp39-abi3-win_amd64.whl", marker = "sys_platform == 'win32'", extra = "attention" },
 ]
 
 [[tool.uv.dependency-metadata]]