From 49ed4adff593447e5c93a06bb385af38979ba526 Mon Sep 17 00:00:00 2001
From: Jianan Xing <1633398+xingjianan@users.noreply.github.com>
Date: Mon, 19 Aug 2024 16:46:28 -0700
Subject: [PATCH 1/2] Relax triton requirements for compatibility with pytorch
 2.4 and newer

Similar to https://github.com/openai/whisper/pull/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
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 183b52755..73c4eb831 100644
--- 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",

From de8eb63f09f1b44cbf29dd870f9598f9e476c5ca Mon Sep 17 00:00:00 2001
From: Jianan Xing <1633398+xingjianan@users.noreply.github.com>
Date: Mon, 19 Aug 2024 16:49:40 -0700
Subject: [PATCH 2/2] Update requirements.txt

---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 62f5f9d7c..8ee592008 100644
--- 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"