Update pyproject.toml

This commit is contained in:
ImpactFrames
2025-03-09 20:02:25 +00:00
committed by GitHub
parent c6212870c7
commit ef7b214cfe
+7
View File
@@ -25,6 +25,13 @@ dependencies = [
[project.optional-dependencies]
# triton
'triton; sys_platform == "linux"',
# triton for windows
'https://github.com/woct0rdho/triton-windows/releases/download/v3.1.0-windows.post8/triton-3.1.0-cp312-cp312-win_amd64.whl; sys_platform == "win64" and (python_version >= "3.12" and python_version < "3.13")',
'https://github.com/woct0rdho/triton-windows/releases/download/v3.1.0-windows.post8/triton-3.1.0-cp311-cp311-win_amd64.whl; sys_platform == "win64" and (python_version >= "3.11" and python_version < "3.12")',
'https://github.com/woct0rdho/triton-windows/releases/download/v3.1.0-windows.post8/triton-3.1.0-cp310-cp310-win_amd64.whl; sys_platform == "win64" and (python_version >= "3.10" and python_version < "3.11")',
'https://github.com/woct0rdho/triton-windows/releases/download/v3.1.0-windows.post8/triton-3.1.0-cp38-cp38-win_amd64.whl; sys_platform == "win64" and (python_version >= "3.8" and python_version < "3.9")',
#flash-attn = ["flash-attn>=2.0.0"]
[project.urls]