Try installing triton only if linux & x86_64 (openai#1051)

ilanit1997 · May 16, 2023 · 77a0a3f · 77a0a3f
1 parent 2689b60
commit 77a0a3f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,4 +1,5 @@
 import os
+import platform
 import sys
 
 import pkg_resources
@@ -11,8 +12,8 @@ def read_version(fname="whisper/version.py"):
 
 
 requirements = []
-if sys.platform.startswith("linux"):
-    triton_requirement = "triton>=2.0.0.dev20221202"
+if sys.platform.startswith("linux") and platform.machine() == "x86_64":
+    triton_requirement = "triton==2.0.0"
     try:
         import re
         import subprocess