-
Notifications
You must be signed in to change notification settings - Fork 178
/
setup.py
47 lines (41 loc) · 1.2 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
import os
from setuptools import setup, find_packages
def version() -> str:
with open(os.path.join(os.path.dirname(__file__), 'stable_whisper/_version.py')) as f:
return f.read().split('=')[-1].strip().strip('"').strip("'")
def read_me() -> str:
with open('README.md', 'r', encoding='utf-8') as f:
return f.read()
setup(
name="stable-ts",
version=version(),
description="Modifies OpenAI's Whisper to produce more reliable timestamps.",
long_description=read_me(),
long_description_content_type='text/markdown',
python_requires=">=3.8",
author="Jian",
url="https://github.com/jianfch/stable-ts",
license="MIT",
packages=find_packages(include=["stable_whisper", "stable_whisper.*"]),
install_requires=[
"numpy",
"torch",
"torchaudio",
"tqdm",
"openai-whisper>=20230314,<=20240930"
],
extras_require={
"fw": [
"faster-whisper"
],
"hf": [
"transformers>=4.23.0",
"optimum",
"accelerate"
]
},
entry_points={
"console_scripts": ["stable-ts=stable_whisper.whisper_word_level:cli"],
},
include_package_data=False
)