cog.yaml

# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md
build:
  gpu: true
  python_version: "3.8"
  python_packages:
    - "accelerate==0.25.0"
    - "diffusers==0.21.0"
    - "torch==2.0.1"
    - "pytorch_lightning==2.1.3"
    - "transformers==4.25.0"
    - "xformers"
    - "loralib==0.1.2"
    - "fairscale==0.4.13"
    - "opencv-python==4.9.0.80"
    - "chardet==5.2.0"
    - "einops==0.7.0"
    - "scipy==1.10.1"
    - "timm==0.9.12"

  run:
    - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.3.1/pget" && chmod +x /usr/local/bin/pget
# predict.py defines how predictions are run on your model
predict: "predict.py:Predictor"