Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-23 22:15:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix triton env marker (#1887)

Browse Source
This commit is contained in:
Bob Lin 2023-12-11 23:39:08 +08:00 committed by GitHub
parent e58f288045
commit 8bc8860694
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
requirements.txt
View File

@ -4,3 +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"

7
setup.py
View File

@ -1,6 +1,6 @@
import os
import platform import platform
import sys import sys
from pathlib import Path
import pkg_resources import pkg_resources
from setuptools import find_packages, setup from setuptools import find_packages, setup
@ -28,11 +28,10 @@ setup(
url="https://github.com/openai/whisper", url="https://github.com/openai/whisper",
license="MIT", license="MIT",
packages=find_packages(exclude=["tests*"]), packages=find_packages(exclude=["tests*"]),
install_requires=requirements install_requires=[
+ [
str(r) str(r)
for r in pkg_resources.parse_requirements( for r in pkg_resources.parse_requirements(
open(os.path.join(os.path.dirname(__file__), "requirements.txt")) Path(__file__).with_name("requirements.txt").open()
) )
], ],
entry_points={ entry_points={