from setuptools import setup, find_packages

setup(
    name='stable-audio-tools',
    version='0.0.19',
    url='https://github.com/Stability-AI/stable-audio-tools.git',
    author='Stability AI',
    description='Training and inference tools for generative audio models from Stability AI',
    packages=find_packages(),  
    install_requires=[
        'alias-free-torch==0.0.6',
        'auraloss==0.4.0',
        'descript-audio-codec==1.0.0',
        'einops',
        'einops-exts',
        'ema-pytorch==0.2.3',
        'encodec==0.1.1',
        'gradio>=5.20.0',
        'huggingface_hub',
        'importlib-resources==5.12.0',
        'k-diffusion==0.1.1',
        'laion-clap==1.1.4',
        'local-attention==1.8.6',
        'pandas==2.0.2',
        'prefigure==0.0.9',
        'pytorch_lightning==2.1.0',
        'PyWavelets==1.4.1',
        'safetensors',
        'sentencepiece==0.1.99',
        'torch>=2.5,<=2.8',
        'torchaudio>=2.5,<=2.8',
        'torchmetrics==0.11.4',
        'tqdm',
        'transformers',
        'v-diffusion-pytorch==0.0.2',
        'vector-quantize-pytorch==1.14.41',
        'wandb==0.15.4',
        'webdataset==0.2.100'
    ],
)