|
|
"""Setup script for Chiluka TTS package.""" |
|
|
|
|
|
from setuptools import setup, find_packages |
|
|
|
|
|
with open("README.md", "r", encoding="utf-8") as fh: |
|
|
long_description = fh.read() |
|
|
|
|
|
setup( |
|
|
name="chiluka", |
|
|
version="0.1.0", |
|
|
author="Seemanth", |
|
|
author_email="seemanth.k@purviewservices.com", |
|
|
description="Chiluka - A lightweight TTS inference package based on StyleTTS2", |
|
|
long_description=long_description, |
|
|
long_description_content_type="text/markdown", |
|
|
url="https://github.com/PurviewVoiceBot/chiluka", |
|
|
packages=find_packages(), |
|
|
include_package_data=False, |
|
|
package_data={ |
|
|
"chiluka": [ |
|
|
"configs/*.yml", |
|
|
"pretrained/ASR/config.yml", |
|
|
"pretrained/ASR/*.py", |
|
|
"pretrained/JDC/*.py", |
|
|
"pretrained/PLBERT/config.yml", |
|
|
"pretrained/PLBERT/*.py", |
|
|
"models/*.py", |
|
|
"models/diffusion/*.py", |
|
|
], |
|
|
}, |
|
|
exclude_package_data={ |
|
|
"chiluka": [ |
|
|
"checkpoints/*.pth", |
|
|
"pretrained/ASR/*.pth", |
|
|
"pretrained/JDC/*.t7", |
|
|
"pretrained/PLBERT/*.t7", |
|
|
], |
|
|
}, |
|
|
classifiers=[ |
|
|
"Development Status :: 3 - Alpha", |
|
|
"Intended Audience :: Developers", |
|
|
"Intended Audience :: Science/Research", |
|
|
"License :: OSI Approved :: MIT License", |
|
|
"Operating System :: OS Independent", |
|
|
"Programming Language :: Python :: 3", |
|
|
"Programming Language :: Python :: 3.8", |
|
|
"Programming Language :: Python :: 3.9", |
|
|
"Programming Language :: Python :: 3.10", |
|
|
"Programming Language :: Python :: 3.11", |
|
|
"Topic :: Multimedia :: Sound/Audio :: Speech", |
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence", |
|
|
], |
|
|
python_requires=">=3.8", |
|
|
install_requires=[ |
|
|
"torch>=1.13.0", |
|
|
"torchaudio>=0.13.0", |
|
|
"transformers>=4.20.0", |
|
|
"librosa>=0.9.0", |
|
|
"phonemizer>=3.0.0", |
|
|
"nltk>=3.7", |
|
|
"PyYAML>=6.0", |
|
|
"munch>=2.5.0", |
|
|
"einops>=0.6.0", |
|
|
"einops-exts>=0.0.4", |
|
|
"numpy>=1.21.0", |
|
|
"scipy>=1.7.0", |
|
|
"huggingface_hub>=0.16.0", |
|
|
], |
|
|
extras_require={ |
|
|
"playback": ["pyaudio>=0.2.11"], |
|
|
"dev": [ |
|
|
"pytest>=7.0.0", |
|
|
"black>=22.0.0", |
|
|
"isort>=5.10.0", |
|
|
], |
|
|
}, |
|
|
entry_points={ |
|
|
"console_scripts": [ |
|
|
"chiluka=chiluka.cli:main", |
|
|
], |
|
|
}, |
|
|
) |
|
|
|
