File size: 2,962 Bytes
6d9dd36 | 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 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
import platform
import os
from setuptools import find_packages, setup
# add fused `anti_alias_activation` cuda extension if CUDA is available
anti_alias_activation_cuda_ext = None
if platform.system() != "Darwin":
try:
from torch.utils import cpp_extension
if cpp_extension.CUDA_HOME is not None:
anti_alias_activation_cuda_ext = cpp_extension.CUDAExtension(
name="indextts.BigVGAN.alias_free_activation.cuda.anti_alias_activation_cuda",
sources=[
"indextts/BigVGAN/alias_free_activation/cuda/anti_alias_activation.cpp",
"indextts/BigVGAN/alias_free_activation/cuda/anti_alias_activation_cuda.cu",
],
include_dirs=["indextts/BigVGAN/alias_free_activation/cuda"],
extra_compile_args={
"cxx": ["-O3"],
"nvcc": [
"-O3",
"--use_fast_math",
"-U__CUDA_NO_HALF_OPERATORS__",
"-U__CUDA_NO_HALF_CONVERSIONS__",
"--expt-relaxed-constexpr",
"--expt-extended-lambda",
],
},
)
else:
print("CUDA_HOME is not set. Skipping anti_alias_activation CUDA extension.")
except ImportError:
print("PyTorch is not installed. Skipping torch extension.")
setup(
name="indextts",
version="0.1.4",
author="Index SpeechTeam",
author_email="xuanwu@bilibili.com",
long_description=open("README.md", encoding="utf8").read(),
long_description_content_type="text/markdown",
description="An Industrial-Level Controllable and Efficient Zero-Shot Text-To-Speech System",
url="https://github.com/index-tts/index-tts",
packages=find_packages(),
include_package_data=True,
install_requires=[
"torch>=2.1.2",
"torchaudio",
"transformers==4.36.2",
"accelerate",
"tokenizers==0.15.0",
"einops==0.8.1",
"matplotlib==3.8.2",
"omegaconf",
"sentencepiece",
"librosa",
"numpy",
"wetext" if platform.system() == "Darwin" else "WeTextProcessing",
],
extras_require={
"webui": ["gradio"],
},
ext_modules=[anti_alias_activation_cuda_ext] if anti_alias_activation_cuda_ext else [],
cmdclass={"build_ext": cpp_extension.BuildExtension} if anti_alias_activation_cuda_ext else {},
entry_points={
"console_scripts": [
"indextts = indextts.cli:main",
]
},
license="Apache-2.0",
python_requires=">=3.10",
classifiers=[
"Programming Language :: Python :: 3.10",
"Operating System :: OS Independent",
"Intended Audience :: Science/Research",
"Topic :: Scientific/Engineering",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
) |