# Configuration for Cog ⚙️
# https://replicate.com/docs/guides/push-a-model
# prerequisite：https://docs.docker.com/engine/install/ubuntu/#set-up-the-repository  `dockerd` to start docker
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md
# !!!! recommend 60G disk space for cog docker

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

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

  # python version in the form '3.8' or '3.8.12'
  python_version: "3.10.4"

  # a list of packages in the format <package-name>==<version>
  python_packages:
    - blendmodes==2022
    - transformers==4.25.1
    - accelerate==0.12.0
    - basicsr==1.4.2
    - gfpgan==1.3.8
    - gradio==3.16.2
    - numpy==1.23.3
    - Pillow==9.4.0
    - realesrgan==0.3.0
    # - torch==1.13.1+cu117
    # - --extra-index-url https://download.pytorch.org/whl/cu117
    # - torchvision==0.14.1+cu117
    # - --extra-index-url https://download.pytorch.org/whl/cu117
    - omegaconf==2.2.3
    - pytorch_lightning==1.7.6
    - scikit-image==0.19.2
    - fonts
    - font-roboto
    - timm==0.6.7
    - piexif==1.1.3
    - einops==0.4.1
    - jsonmerge==1.8.0
    - clean-fid==0.1.29
    - resize-right==0.0.2
    - torchdiffeq==0.2.3
    - kornia==0.6.7
    - lark==1.1.2
    - inflection==0.5.1
    - GitPython==3.1.27
    - torchsde==0.2.5
    - safetensors==0.2.7
    - httpcore<=0.15
    - fastapi==0.90.1
    # - open_clip_torch
    - git+https://github.com/mlfoundations/open_clip.git@bb6e834e9c70d9c27d0dc3ecedeebeaeb1ffad6b
    - git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1
  
  # commands run after the environment is setup
  run:
    - "pip3 install torch==1.13.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117"
    - "pip3 install torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117"
    - "echo env is ready!"

# https://replicate.com/wolverinn/chill_watcher
image: "r8.im/wolverinn/chill_watcher"

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