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'",