Relax triton requirements for compatibility with pytorch 2.1 and newer

openai · Nov 13, 2023 · 2302f2d · 2302f2d
1 parent fcfeaf1
commit 2302f2d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -13,7 +13,7 @@ def read_version(fname="whisper/version.py"):
 
 requirements = []
 if sys.platform.startswith("linux") and platform.machine() == "x86_64":
-    requirements.append("triton==2.0.0")
+    requirements.append("triton>=2.0.0,<3")
 
 setup(
     name="openai-whisper",