Relax triton requirements for compatibility with pytorch 2.1 and newer (#1802)

This commit is contained in:
Eugene Indenbom
2023-11-13 19:43:42 +02:00
committed by GitHub
parent fcfeaf1b61
commit 1cea435768

View File

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