From ca96c87e444213ea53b966d8c45b2d66af1f9f8c Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Wed, 16 Jul 2025 21:00:27 -0700
Subject: [PATCH] fix sageattention dependency

---
 pyproject.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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]]