Relax triton requirements for compatibility with pytorch 2.4 and newer (

openai#2307) * Relax triton requirements for compatibility with pytorch 2.4 and newer Similar to openai#1802, but now when pytorch upgrades to 2.4, it requires triton==3.0.0. I am not sure if it makes sense to remove the upper bound version constraints * Update requirements.txt
sdifiore · Sep 10, 2024 · 32d55d5 · 32d55d5
1 parent ba3f3cd
commit 32d55d5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -4,4 +4,4 @@ torch
 tqdm
 more-itertools
 tiktoken
-triton>=2.0.0,<3;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2"
+triton>=2.0.0;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2"
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,<3")
+    requirements.append("triton>=2.0.0")
 
 setup(
     name="openai-whisper",