Skip to content

Commit 1cea435

Browse files
authored
Relax triton requirements for compatibility with pytorch 2.1 and newer (ggml-org#1802)
1 parent fcfeaf1 commit 1cea435

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
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")
16+
requirements.append("triton>=2.0.0,<3")
1717

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

0 commit comments

Comments
 (0)