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

Try installing triton only if linux & x86_64 (#1051)

Browse Source
This commit is contained in:
Jong Wook Kim 2023-03-07 14:31:40 -05:00 committed by GitHub
parent 4b0d5e58d0
commit 924e1f8e06
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
setup.py
View File

@ -1,4 +1,5 @@
import os
import platform
import sys
import pkg_resources
@ -11,8 +12,8 @@ def read_version(fname="whisper/version.py"):
requirements = []
if sys.platform.startswith("linux"):
triton_requirement = "triton>=2.0.0.dev20221202"
if sys.platform.startswith("linux") and platform.machine() == "x86_64":
triton_requirement = "triton==2.0.0"
try:
import re
import subprocess