Audio-Tokenizer-Train/setup.py at main · Mi221e/Audio-Tokenizer-Train · 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
from setuptools import find_packages
from setuptools import setup

with open("README.md") as f:
    long_description = f.read()

setup(
    name="audio-tokenizer-train",
    version="1.0.0",
    classifiers=[
        "Intended Audience :: Developers",
        "Natural Language :: English",
        "Programming Language :: Python :: 3",
        "Topic :: Multimedia :: Sound/Audio",
        "Topic :: Software Development :: Libraries",
    ],
    description="HF-compatible training and inference code for MOSS/CAT audio tokenizers.",
    long_description=long_description,
    long_description_content_type="text/markdown",
    author="",
    author_email="",
    url="https://github.com/Mi221e/Audio-Tokenizer-Train",
    license="MIT",
    packages=find_packages(),
    keywords=["audio", "tokenizer", "codec", "machine learning"],
    install_requires=[
        "argbind>=0.3.7",
        "descript-audiotools>=0.7.2",
        "einops",
        "numpy",
        "PyYAML",
        "torch",
        "torchaudio",
        "tqdm",
        "tensorboard",
        "transformers>=4.51.0",
        "accelerate",
        "safetensors",
        "jiwer",
        "rapidfuzz",
        "soundfile",
        "protobuf<3.20,>=3.19.2",
        "wandb==0.17.9",
    ],
    extras_require={
        "dev": [
            "pytest",
            "pytest-cov",
            "pynvml",
            "psutil",
            "pandas",
            "onnx",
            "onnx-simplifier",
            "seaborn",
            "jupyterlab",
            "watchdog",
            "pesq",
            "tabulate",
            "encodec",
        ],
    },
)