Skip to content

Commit 2b83efd

Browse files
xingjiananhiden47
authored andcommitted
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
1 parent d83c064 commit 2b83efd

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,4 @@ torch
44
tqdm
55
more-itertools
66
tiktoken
7-
triton>=2.0.0,<3;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2"
7+
triton>=2.0.0;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2"

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ def read_version(fname="whisper/version.py"):
1313

1414
requirements = []
1515
if sys.platform.startswith("linux") and platform.machine() == "x86_64":
16-
requirements.append("triton>=2.0.0,<3")
16+
requirements.append("triton>=2.0.0")
1717

1818
setup(
1919
name="openai-whisper",

0 commit comments

Comments
 (0)