From ef7b214cfee2d7f6ee3d704f6dde23979d5257c4 Mon Sep 17 00:00:00 2001 From: ImpactFrames Date: Sun, 9 Mar 2025 20:02:25 +0000 Subject: [PATCH] Update pyproject.toml --- pyproject.toml | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/pyproject.toml b/pyproject.toml index 60b2e39..8d30bd8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -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]