setup.py 692 Bytes
Newer Older
mrfakename's avatar
mrfakename committed
1
2
3
4
5
6
7
8
from setuptools import setup

setup(
    name="tangoflux",
    description="TangoFlux: Super Fast and Faithful Text to Audio Generation with Flow Matching",
    version="0.1.0",
    packages=["tangoflux"],
    install_requires=[
Nuomanzz's avatar
Nuomanzz committed
9
10
11
12
13
        "torch",
        "torchaudio",
        "torchlibrosa",
        "torchvision==",
        "transformers==",
mrfakename's avatar
mrfakename committed
14
        "diffusers==0.30.0",
Nuomanzz's avatar
Nuomanzz committed
15
16
        "accelerate",
        "datasets",
mrfakename's avatar
mrfakename committed
17
18
19
20
21
22
23
24
25
26
27
28
29
30
        "librosa",
        "tqdm",
        "wandb",
        "click",
        "gradio",
        "torchaudio",
    ],
    entry_points={
        "console_scripts": [
            "tangoflux=tangoflux.cli:main",
            "tangoflux-demo=tangoflux.demo:main",
        ],
    },
)