CapSpeech/setup.py at main · WangHelin1997/CapSpeech · GitHub

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
from setuptools import setup, find_packages

setup(
    name='capspeech',
    version='0.1.0',
    packages=find_packages(),
    install_requires=[
        'xformers',
        'g2p_en',
        "laion_clap",
        'torchvision',
        'torchaudio',
        'torch',
        'transformers>=4.46.1,<=4.46.1',
        'tensorboard',
        'datasets',
        'torchmetrics',
        'huggingface_hub',
        'cached_path',
        'torchdiffeq',
        'x_transformers',
        'jieba',
        'pypinyin',
        'wandb',
        'ema_pytorch',
        "sentencepiece",
        "descript-audio-codec",
        "descript-audiotools @ git+https://github.com/descriptinc/audiotools",
        "protobuf>=4.0.0",
        "jiwer",
        "accelerate",
        "evaluate",
        "ninja",
        "packaging",
        "wheel"
    ],
    author='Helin Wang',
    description='CapSpeech: Enabling Downstream Applications in Style-Captioned Text-to-Speech',
    url='https://github.com/WangHelin1997/CapSpeech',
    python_requires='>=3.10',
)