Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-03-30 14:28:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Relax triton requirements for compatibility with pytorch 2.4 and newer (#2307)

Browse Source 
* Relax triton requirements for compatibility with pytorch 2.4 and newer

Similar to https://github.com/openai/whisper/pull/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
This commit is contained in:
Jianan Xing 2024-09-10 09:53:08 -07:00 committed by GitHub
parent ba3f3cd54b
commit 32d55d5d76
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@ -4,4 +4,4 @@ torch
tqdm tqdm
more-itertools more-itertools
tiktoken tiktoken
triton>=2.0.0,<3;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2" triton>=2.0.0;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2"

2
setup.py
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,<3") requirements.append("triton>=2.0.0")
setup( setup(
name="openai-whisper", name="openai-whisper",