chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #541 from justinwlin/main

Browse Source 
Update setup.py to download pyannote depending on platform
This commit is contained in:
Max Bain
2023-10-25 11:14:11 -07:00
committed by GitHub
parent 49e0130e4e b522133340
commit e9a6385d3c
1 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
setup.py
View File

@ -1,8 +1,16 @@
import os import os
import platform
import pkg_resources import pkg_resources
from setuptools import setup, find_packages from setuptools import setup, find_packages
def get_pyannote_audio_version():
machine = platform.machine()
system = platform.system()
version = "3.0.0" if machine == "aarch64" or system == "Darwin" else "3.0.1"
return version
setup( setup(
name="whisperx", name="whisperx",
py_modules=["whisperx"], py_modules=["whisperx"],
@ -19,10 +27,11 @@ setup(
for r in pkg_resources.parse_requirements( for r in pkg_resources.parse_requirements(
open(os.path.join(os.path.dirname(__file__), "requirements.txt")) open(os.path.join(os.path.dirname(__file__), "requirements.txt"))
) )
] + ["pyannote.audio==3.0.1"], ]
entry_points = { + [f"pyannote.audio=={get_pyannote_audio_version()}"],
'console_scripts': ['whisperx=whisperx.transcribe:cli'], entry_points={
"console_scripts": ["whisperx=whisperx.transcribe:cli"],
}, },
include_package_data=True, include_package_data=True,
extras_require={'dev': ['pytest']}, extras_require={"dev": ["pytest"]},
) )