Skip to content

Commit 8bc8860

Browse files
authored
Fix triton env marker (#1887)
1 parent e58f288 commit 8bc8860

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,4 @@ torch
44
tqdm
55
more-itertools
66
tiktoken
7+
triton>=2.0.0,<3;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2"

setup.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
import os
21
import platform
32
import sys
3+
from pathlib import Path
44

55
import pkg_resources
66
from setuptools import find_packages, setup
@@ -28,11 +28,10 @@ def read_version(fname="whisper/version.py"):
2828
url="https://github.com/openai/whisper",
2929
license="MIT",
3030
packages=find_packages(exclude=["tests*"]),
31-
install_requires=requirements
32-
+ [
31+
install_requires=[
3332
str(r)
3433
for r in pkg_resources.parse_requirements(
35-
open(os.path.join(os.path.dirname(__file__), "requirements.txt"))
34+
Path(__file__).with_name("requirements.txt").open()
3635
)
3736
],
3837
entry_points={

0 commit comments

Comments
 (0)