fix sageattention resources on macos

This commit is contained in:
doctorpangloss 2025-07-16 20:47:24 -07:00
parent ff9832e6ba
commit c28a4b9c4a

View File

@ -166,7 +166,7 @@ mps = [
] ]
attention = [ attention = [
"sageattention", "sageattention==2.2.0 ; sys_platform == 'Linux' or sys_platform == 'win32'",
"flash_attn ; sys_platform == 'Linux'", "flash_attn ; sys_platform == 'Linux'",
] ]
@ -244,11 +244,20 @@ torchvision = [
torchaudio = [ torchaudio = [
{ index = "pytorch-cpu", extra = "cpu" }, { index = "pytorch-cpu", extra = "cpu" },
] ]
"sageattention" = [ sageattention = [
{ git = "https://github.com/thu-ml/SageAttention.git", extra = "attention", marker = "sys_platform == 'Linux' or sys_platform == 'win32'" }, { 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] [tool.ruff]
lint.select = [ lint.select = [