| """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", |
| ], |
| }, |
| ) |
|
|
