From c28a4b9c4ad97f7abca0b113cd0bdf54f64d307f Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Wed, 16 Jul 2025 20:47:24 -0700
Subject: [PATCH] fix sageattention resources on macos

---
 pyproject.toml | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 990113452..0c8a9322b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -166,7 +166,7 @@ mps = [
 ]
 
 attention = [
-    "sageattention",
+    "sageattention==2.2.0 ; sys_platform == 'Linux' or sys_platform == 'win32'",
     "flash_attn ; sys_platform == 'Linux'",
 ]
 
@@ -244,11 +244,20 @@ torchvision = [
 torchaudio = [
     { index = "pytorch-cpu", extra = "cpu" },
 ]
-"sageattention" = [
-    { git = "https://github.com/thu-ml/SageAttention.git", extra = "attention", marker = "sys_platform == 'Linux' or sys_platform == 'win32'" },
+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%2Bcu128torch2.8.0-cp38-abi3-win_amd64.whl", marker = "sys_platform == 'win32'" },
 ]
 
+[[tool.uv.dependency-metadata]]
+name = "flash-attn"
+version = "2.6.3"
+requires-dist = ["torch", "einops"]
 
+[[tool.uv.dependency-metadata]]
+name = "sageattention"
+version = "2.2.0"
+requires-dist = ["torch"]
 
 [tool.ruff]
 lint.select = [