cog.yaml

# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md

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

  # a list of ubuntu apt packages to install
  system_packages:
    - "libgl1"

  # python version in the form '3.11' or '3.11.4'
  python_version: "3.10.12"

  # a list of packages in the format <package-name>==<version>
  python_packages:
    - "torch==2.0.1"
    - "torchvision==0.15.2"
    - "xformers==0.0.20"

  # commands run after the environment is setup
  run:
    - git clone --branch v1.7.0 https://github.com/AUTOMATIC1111/stable-diffusion-webui /stable-diffusion-webui
    - git clone https://github.com/Bing-su/adetailer.git /stable-diffusion-webui/extensions/adetailer
    - git clone https://github.com/speshiou/cog-sd-webui /cog-sd-webui
    - wget --content-disposition -P /stable-diffusion-webui/embeddings "https://huggingface.co/datasets/gsdf/EasyNegative/resolve/main/EasyNegative.safetensors?download=true"
    - python /cog-sd-webui/init_env.py --skip-torch-cuda-test

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