build:
  gpu: true
  cuda: "11.8"
  python_version: "3.10"
  python_packages:
    - "audiolm-pytorch==1.1.4"
    - "fairseq"
    - "huggingface-hub"
    - "sentencepiece"
    - "transformers"
    - "encodec"
    - 'soundfile; platform_system == "Windows"'
    - 'sox; platform_system != "Windows"'
    - "tensorboardX"
    - "torch"    
    - "torchvision"
    - "torchaudio"
    - "light-the-torch"  

predict: "predict.py:Predictor"