From 279dc28b1d8ac7a03ca8076e3a2c0e26c87eeea5 Mon Sep 17 00:00:00 2001
From: Bryce <github20210803@accounts.brycedrennan.com>
Date: Wed, 17 Jan 2024 22:15:05 -0800
Subject: [PATCH] build: triton only has wheels for linux

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index e3300f91..a9193f99 100644
--- a/setup.py
+++ b/setup.py
@@ -105,7 +105,7 @@ def get_git_revision_hash() -> str:
         "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'",