From 1cea4357687b676b293cb5473e1ade25f5b1cef7 Mon Sep 17 00:00:00 2001
From: Eugene Indenbom <45334274+eindenbom@users.noreply.github.com>
Date: Mon, 13 Nov 2023 19:43:42 +0200
Subject: [PATCH] Relax triton requirements for compatibility with pytorch 2.1
 and newer (#1802)

---
 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",