From 2302f2d7a2672fc20c22aba9acbd2775e021978a Mon Sep 17 00:00:00 2001
From: Eugene Indenbom <45334274+eindenbom@users.noreply.github.com>
Date: Mon, 13 Nov 2023 15:50:35 +0200
Subject: [PATCH] Relax triton requirements for compatibility with pytorch 2.1
 and newer

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 1161d815a..ae8589e1d 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")
+    requirements.append("triton>=2.0.0,<3")
 
 setup(
     name="openai-whisper",