cog.yaml

# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md

build:
  gpu: true
  cuda: "12.1"

  system_packages:
    - ffmpeg

  python_version: "3.10"

  python_packages:
    - 'aeiou==0.0.20'
    - 'alias-free-torch==0.0.6'
    - 'auraloss==0.4.0'
    - 'descript-audio-codec==1.0.0'
    - 'einops==0.7.0'
    - 'einops-exts==0.0.4'
    - 'ema-pytorch==0.2.3'
    - 'encodec==0.1.1'
    - 'gradio>=3.42.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'
    - 'pedalboard==0.7.4'
    - 'prefigure==0.0.9'
    - 'pytorch_lightning==2.1.0' 
    - 'PyWavelets==1.4.1'
    - 'safetensors'
    - 'sentencepiece==0.1.99'
    - 's3fs'
    - 'torch>=2.0.1'
    - 'torchaudio>=2.0.2'
    - 'torchmetrics==0.11.4'
    - 'tqdm'
    - 'transformers'
    - 'v-diffusion-pytorch==0.0.2'
    - 'vector-quantize-pytorch==1.9.14'
    - 'wandb==0.15.4'
    - 'webdataset==0.2.48'
    - 'x-transformers<1.27.0'
    - 'setuptools==69.5.1'

  run:
    - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.6.2/pget_linux_x86_64" && chmod +x /usr/local/bin/pget

predict: "predict.py:Predictor"