cog.yaml

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

build:
  # set to true if your model requires a GPU
  gpu: true
  cuda: "12.1"

  # a list of ubuntu apt packages to install
  # system_packages:
  #   - "libgl1-mesa-glx"
  #   - "libglib2.0-0"

  # python version in the form '3.11' or '3.11.4'
  python_version: "3.11"

  # a list of packages in the format <package-name>==<version>
  python_packages:
    - accelerate==0.21.0
    - av==11.0.0
    - decord==0.6.0
    - diffusers==0.24.0
    - einops==0.4.1
    - gradio<=3.50.2
    - imageio==2.33.0
    - imageio-ffmpeg==0.4.9
    - numpy>=1.23.5
    - omegaconf==2.2.3
    - onnxruntime-gpu==1.16.3
    - open-clip-torch==2.20.0
    - opencv-contrib-python==4.8.1.78
    - opencv-python==4.8.1.78
    - Pillow==9.5.0
    - scikit-image==0.21.0
    - scikit-learn==1.3.2
    - scipy==1.11.4
    - torch==2.0.1
    - torchdiffeq==0.2.3
    - torchmetrics==1.2.1
    - torchsde==0.2.5
    - torchvision==0.15.2
    - tqdm==4.66.1
    - transformers==4.30.2
    - mlflow==2.9.2
    - xformers==0.0.22
    - controlnet-aux==0.0.7
    - https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip

  # commands run after the environment is setup
  run:
    - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.5.6/pget_linux_x86_64" && chmod +x /usr/local/bin/pget

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