build:
  gpu: true
  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"
  python_version: "3.10.4"
  python_packages:
    - "torch==2.0.1"
    - "torchvision==0.15.2"
    - "xformers==0.0.22"
    - "tensorboard==2.16.2"
    - "gfpgan==1.3.8"
    - "lpips==0.1.4"
    - "realesrgan==0.3.0"
    - "gdown==5.1.0"
    - "mediapipe==0.10.14"
  run:
    - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/latest/download/pget_$(uname -s)_$(uname -m)"
    - chmod +x /usr/local/bin/pget
    - git config --global --add safe.directory /src
    - git config --global --add safe.directory /src/extensions/sd-webui-controlnet
    - git config --global --add safe.directory /src/extensions/multidiffusion-upscaler-for-automatic1111
    - git clone https://github.com/philz1337x/stable-diffusion-webui-cog-init /stable-diffusion-webui
    - python /stable-diffusion-webui/init_env.py --skip-torch-cuda-test
    - sed -i 's/from pkg_resources import packaging/import packaging/g' /root/.pyenv/versions/3.10.4/lib/python3.10/site-packages/clip/clip.py
predict: "predict.py:Predictor"