Fix triton env marker (#1887)

openai · Dec 11, 2023 · 8bc8860 · n13l420 · May 23, 2024 · 8bc8860
1 parent e58f288
commit 8bc8860
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -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"
diff --git a/setup.py b/setup.py
@@ -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 @@ def read_version(fname="whisper/version.py"):
     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={