# Configuration for Cog ⚙️
# Reference: https://cog.run/yaml

build:
  gpu: true
  cuda: "12.1"
  system_packages:
    - "ffmpeg"
    - "libgl1"
  python_version: "3.10.13"
  python_packages:
    - "torch==2.2.2"
    - "torchvision"
    - "triton==2.2.0"
    - "diffusers==0.11.1"
    - "transformers==4.38.0"
    - "huggingface-hub==0.25.2"
    - "imageio==2.27.0"
    - "decord==0.6.0"
    - "accelerate==0.26.1"
    - "einops==0.7.0"
    - "omegaconf==2.3.0"
    - "safetensors==0.4.2"
    - "opencv-python==4.9.0.80"
    - "mediapipe==0.10.11"
    - "av==11.0.0"
    - "torch-fidelity==0.3.0"
    - "torchmetrics==1.3.1"
    - "python_speech_features==0.6"
    - "librosa==0.10.1"
    - "scenedetect==0.6.1"
    - "ffmpeg-python==0.2.0"
    - "lpips==0.1.4"
    - "face-alignment==1.4.1"
    - "ninja==1.11.1.1"
    - "pandas==2.0.3"
    - "numpy==1.24.4"
    - "xformers==0.0.26"

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

# predict.py defines how predictions are run on your model
predict: "predict.py:Predictor"