from setuptools import setup, find_packages

setup(
    name="mmaudio",
    version="0.1.0",
    description="MMAudio package for audio generation",
    packages=find_packages(),
    install_requires=[
        "torch>=2.1.0",
        "torchaudio>=2.1.0",
        "numpy>=1.21.0",
        "librosa>=0.8.1",
        "soundfile",
        "av",
        "einops>=0.6",
        "hydra-core>=1.3.2",
        "requests",
        "torchdiffeq",
        "nitrous-ema",
        "auraloss",
        "hydra_colorlog",
        "tensordict",
        "colorlog",
        "open_clip_torch",
    ],
    python_requires=">=3.8",
)