# Configuration for Cog ⚙️
# Reference: https://cog.run/yaml

build:
  gpu: true
  cuda: "12.1"
  python_version: "3.11"
  python_packages:
    - "torch==2.2"
    - "torchvision"
    - "transformers==4.36.2"
    - "accelerate==1.3.0"
    - "diffusers==0.32.2"
    - "opencv-python==4.10.0.84"
    - "attrdict==2.0.1"
    - "timm==1.0.14"
    - "sentencepiece==0.2.0"
    - "einops==0.8.0"
    - "pillow==10.2.0"
    - "numpy==1.24.3"

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

predict: "predict.py:Predictor"