versa/setup.py at main · wavlab-speech/versa · 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
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
from setuptools import setup, find_packages

setup(
    name="versa-speech-audio-toolkit",
    version="1.0.0",
    packages=find_packages(),
    install_requires=[
        "accelerate",
        "audioread",
        "ci-sdr",
        "Cython",
        "Distance",
        "editdistance",
        "einops",
        "espnet @ git+https://github.com/ftshijt/espnet.git@espnet_inference#egg=espnet",
        "espnet-tts-frontend",
        "fast-bss-eval",
        "fastdtw",
        "huggingface-hub",
        "hydra-core",
        "idna",
        "importlib-metadata",
        "kaggle",
        "kaldiio",
        "lazy_loader",
        "Levenshtein",
        "librosa",
        "mir-eval",
        "omegaconf",
        "onnxruntime",
        # NOTE(jiatong): use the latest commit for python 3.13
        "openai-whisper @ git+https://github.com/openai/whisper.git",
        "opt-einsum",
        "pesq",
        "protobuf",
        "pysptk",
        "pystoi",
        "python-dateutil",
        "pyworld",
        "pyyaml",
        "rapidfuzz",
        "resampy",
        "safetensors",
        "scikit-learn",
        "sentencepiece",
        "setuptools",
        "soundfile",
        "speechmos",
        "sympy",
        "threadpoolctl",
        "tokenizers",
        "torch",
        "torch-complex",
        "torchaudio",
        "torchlibrosa",
        "s3prl @ git+https://github.com/ftshijt/s3prl.git@numpy2#egg=s3prl",
        "transformers>=4.36.2",
        "espnet_model_zoo",
        "discrete-speech-metrics @ git+https://github.com/ftshijt/DiscreteSpeechMetrics.git@v1.0.2",
        "cdpam",
    ],
    extras_require={
        "dev": [
            "pytest>=6.0.0",
            "pytest-cov>=2.10.0",
            "black>=22.3.0",
            "flake8>=4.0.0",
        ],
    },
    entry_points={
        "console_scripts": [
            "versa-score=versa.bin.scorer:main",
        ],
    },
    author="Jiatong Shi",
    author_email="ftshijt@gmail.com",
    description="A package for versatile evaluation of speech and audio",
    url="https://github.com/shinjiwlab/versa.git",
    keywords="speech metrics",
)