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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -13,7 +13,7 @@ def read_version(fname="whisper/version.py"):
requirements = [] requirements = []
if sys.platform.startswith("linux") and platform.machine() == "x86_64": if sys.platform.startswith("linux") and platform.machine() == "x86_64":
requirements.append("triton==2.0.0") requirements.append("triton>=2.0.0,<3")
setup( setup(
name="openai-whisper", name="openai-whisper",