Update pyproject.toml
This commit is contained in:
@@ -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]
|
||||
|
||||
Reference in New Issue
Block a user