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

abyesilyurt · Nov 13, 2023 · e4402ab · e4402ab
1 parent f9d8e1b
commit e4402ab
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