# Configuration for Cog ⚙️

# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md

build:

gpu: true

system_packages:

- "libgl1-mesa-glx"

- "libglib2.0-0"

python_version: "3.11"

python_packages:

- torch==2.0.1

- torchvision==0.15.2

- diffusers==0.21.2

- transformers==4.25.1

- accelerate==0.23.0

- imageio==2.27.0

- decord==0.6.0

- einops

- omegaconf

- safetensors

- wandb

- moviepy

- scikit-learn

- av

- rotary_embedding_torch

- torchmetrics

- torch-fidelity

run:

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

predict: "predict.py:Predictor"