Fix triton env marker (#1887)

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

View File

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

View File

@ -1,6 +1,6 @@
import os
import platform
import sys
from pathlib import Path
import pkg_resources
from setuptools import find_packages, setup
@ -28,11 +28,10 @@ setup(
url="https://github.com/openai/whisper",
license="MIT",
packages=find_packages(exclude=["tests*"]),
install_requires=requirements
+ [
install_requires=[
str(r)
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={