feat: get rid of pyannote versioning and go to 3.1

This commit is contained in:
Remc
2023-11-17 11:03:23 +01:00
parent ba30365344
commit 1d7f8ccbf1

View File

@ -1,15 +1,8 @@
import os import os
import platform import platform
import pkg_resources import pkg_resources
from setuptools import setup, find_packages from setuptools import find_packages, setup
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",
@ -28,7 +21,7 @@ setup(
open(os.path.join(os.path.dirname(__file__), "requirements.txt")) open(os.path.join(os.path.dirname(__file__), "requirements.txt"))
) )
] ]
+ [f"pyannote.audio=={get_pyannote_audio_version()}"], + [f"pyannote.audio==3.1.0"],
entry_points={ entry_points={
"console_scripts": ["whisperx=whisperx.transcribe:cli"], "console_scripts": ["whisperx=whisperx.transcribe:cli"],
}, },