cog.yaml

build:
  gpu: true
  python_version: "3.10"
  cuda: "11.7"
  python_packages:
    - "accelerate==0.20.3"
    - "diffusers==0.17.1"
    - "gradio==3.35.2"
    - "imageio[ffmpeg]==2.31.1"
    - "torch==2.0.1"
    - "torchvision==0.15.2"
    - "transformers==4.30.2"
    - "einops==0.6.1"
    - "omegaconf==2.3.0"
    - "opencv-python-headless==4.7.0.72"
    - "decord==0.6.0"

predict: "predict.py:Predictor"