mirror of
https://github.com/openai/whisper.git
synced 2025-11-23 22:15:58 +00:00
Relax triton requirements for compatibility with pytorch 2.1 and newer (#1802)
This commit is contained in:
parent
fcfeaf1b61
commit
1cea435768
2
setup.py
2
setup.py
@ -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",
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user