build: triton only has wheels for linux

4 months ago · 279dc28b1d
parent 3906072191
commit 279dc28b1d
1 changed files with 1 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@ -105,7 +105,7 @@ setup(
        "torchdiffeq>=0.2.0",
        "torchvision>=0.13.1",
        "transformers>=4.19.2",
-        "triton>=2.0.0; sys_platform!='darwin' and platform_machine!='aarch64'",
+        "triton>=2.0.0; sys_platform!='darwin' and platform_machine!='aarch64' and sys_platform == 'linux'",
        "kornia>=0.6",
        "uvicorn>=0.16.0",
        # "xformers>=0.0.22; sys_platform!='darwin' and platform_machine!='aarch64'",