cog.yaml

# 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
    - accelerate==0.23.0
    - diffusers==0.17.1
    - einops==0.7.0
    - opencv_python==4.5.5.64
    - packaging==23.2
    - Pillow==10.1.0
    - safetensors==0.4.0
    - tqdm==4.65.0
    - transformers==4.34.1
    - lpips==0.1.4
  run:
    - mkdir -p /root/.cache/torch/hub/checkpoints/ && wget --output-document "/root/.cache/torch/hub/checkpoints/alexnet-owt-7be5be79.pth" "https://download.pytorch.org/models/alexnet-owt-7be5be79.pth"
predict: "predict.py:Predictor"