cog.yaml

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

build:
  gpu: true
  python_version: "3.11"
  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"
  python_packages:
    - "torch==2.0.1"
    - "albumentations==0.4.3"
    - "datasets==2.8.0"
    - "diffusers==0.21.4"
    - "opencv-python==4.8.1.78"
    - "invisible-watermark"
    - "imageio==2.9.0"
    - "imageio-ffmpeg==0.4.2"
    - "pytorch-lightning==1.4.2"
    - "omegaconf==2.1.1"
    - "tokenizers==0.14.1"
    - "einops==0.3.0"
    - "torch-fidelity==0.3.0"
    - "transformers==4.34.0"
    - "torchmetrics==0.6.0"
    - "kornia==0.6"
    - "openai==0.28.1"
    - "k-diffusion==0.1.0"
    - "pycocotools==2.0.7"
  run:
    - pip install git+https://github.com/cocodataset/panopticapi.git
    - pip install -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
    - pip install -e git+https://github.com/openai/CLIP.git@main#egg=clip
predict: "predict.py:Predictor"