.devscripts/chmod.sh

@@ -1,6 +0,0 @@
-#!/bin/bash
-
-set -Eeuo pipefail
-
-find services -name "*.sh" -exec git update-index --chmod=+x {} \;
-find .devscripts -name "*.sh" -exec git update-index --chmod=+x {} \;

.devscripts/migratev1tov2.sh

@@ -1,30 +0,0 @@
-mkdir -p data/.cache data/StableDiffusion data/Codeformer data/GFPGAN data/ESRGAN data/BSRGAN data/RealESRGAN data/SwinIR data/LDSR data/embeddings
-
-cp -vf cache/models/model.ckpt data/StableDiffusion/model.ckpt
-
-cp -vf cache/models/LDSR.ckpt data/LDSR/model.ckpt
-cp -vf cache/models/LDSR.yaml data/LDSR/project.yaml
-
-cp -vf cache/models/RealESRGAN_x4plus.pth data/RealESRGAN/
-cp -vf cache/models/RealESRGAN_x4plus_anime_6B.pth data/RealESRGAN/
-
-cp -vrf cache/torch data/.cache/
-
-mkdir -p data/.cache/huggingface/transformers/
-cp -vrf cache/transformers/* data/.cache/huggingface/transformers/
-
-cp -v cache/custom-models/* data/StableDiffusion/
-
-mkdir -p data/.cache/clip/
-cp -vf cache/weights/ViT-L-14.pt data/.cache/clip/
-
-cp -vf cache/weights/codeformer.pth data/Codeformer/codeformer-v0.1.0.pth
-
-cp -vf cache/weights/detection_Resnet50_Final.pth data/.cache/
-cp -vf cache/weights/parsing_parsenet.pth data/.cache/
-
-cp -v embeddings/* data/embeddings/
-
-echo this script was created 10/2022
-echo Dont forget to run: docker compose --profile download up --build
-echo the cache and embeddings folders can be deleted, but its not necessary.

.devscripts/migratev3tov4.sh

@@ -1,9 +0,0 @@
-#!/bin/bash
-
-set -Eeuo pipefail
-
-echo "Moving everything in output to output/old..."
-mv output old
-mkdir output
-mv old/.gitignore output
-mv old output

.devscripts/migratev7tov8.sh

@@ -1,28 +0,0 @@
-#!/bin/bash
-
-set -Eeuo pipefail
-
-echo "Renaming..."
-
-# compatible with default auto-names
-mv -v ./data/StableDiffusion ./data/Stable-diffusion
-mv -v ./data/Deepdanbooru ./data/torch_deepdanbooru
-
-# casing problem on windows
-mv -v ./data/Hypernetworks ./data/hypernetworks1
-mv -v ./data/hypernetworks1 ./data/hypernetworks
-
-mv -v ./data/MiDaS ./data/midas1
-mv -v ./data/midas1 ./data/midas
-
-
-echo "Moving folders..."
-
-mkdir -pv ./final
-
-mv -v ./data/config ./final/config
-mv -v ./data/.cache ./final/.cache
-mv -v ./data/embeddings ./final/embeddings
-mv -v ./data ./final/models
-
-mv -v ./final ./data

.editorconfig

@@ -1,9 +0,0 @@
-root = true
-
-[*]
-end_of_line = lf
-indent_style = space
-indent_size = 2
-charset = utf-8
-insert_final_newline = true
-trim_trailing_whitespace = true

.gitattributes

@@ -1 +0,0 @@
-* text=auto eol=lf

.github/ISSUE_TEMPLATE/bug.md

@@ -1,43 +0,0 @@
----
-name: Bug
-about: Report a bug
-title: ""
-labels: bug
-assignees: ""
----
-
-<!--  PLEASE FILL THIS OUT, IT WILL MAKE BOTH OF OUR LIVES EASIER -->
-
-**Has this issue been opened before?**
-
-- [ ] It is not in the [FAQ](https://github.com/AbdBarho/stable-diffusion-webui-docker/wiki/FAQ), I checked.
-- [ ] It is not in the [issues](https://github.com/AbdBarho/stable-diffusion-webui-docker/issues?q=), I searched.
-
-**Describe the bug**
-
-<!--  tried to run the app, my cat exploded -->
-
-**Which UI**
-
-auto or auto-cpu or invoke or comfy?
-
-**Hardware / Software**
-
-- OS: [e.g. Windows 10 / Ubuntu 22.04]
-- OS version: <!--  on windows, use the command `winver` to find out, on ubuntu `lsb_release -d`  -->
-- WSL version (if applicable): <!-- get using `wsl -l -v` -->
-- Docker Version: <!--  get using `docker version` -->
-- Docker compose version: <!--  get using `docker compose version` -->
-- Repo version: <!-- tag, commit sha, or "from master" -->
-- RAM:
-- GPU/VRAM:
-
-**Steps to Reproduce**
-
-1. Go to '...'
-2. Click on '....'
-3. Scroll down to '....'
-4. See error
-
-**Additional context**
-Any other context about the problem here. If applicable, add screenshots to help explain your problem.

.github/ISSUE_TEMPLATE/config.yml

@@ -1,5 +0,0 @@
-blank_issues_enabled: false
-contact_links:
-  - name: Feature request? Questions regarding some extension?
-    url: https://github.com/AbdBarho/stable-diffusion-webui-docker/discussions
-    about: Please use the discussions tab

.github/pull_request_template.md

@@ -1,13 +0,0 @@
-<!--
-Have you created an issue before opening a merge request???
-https://github.com/AbdBarho/stable-diffusion-webui-docker#contributing
-Please create one so we can discuss it, I don't want your effort to go to waste.
--->
-
-Closes issue #
-
-### Update versions
-
-- auto: https://github.com/AUTOMATIC1111/stable-diffusion-webui/commit/
-- invoke: https://github.com/invoke-ai/InvokeAI/commit/
-- comfy: https://github.com/comfyanonymous/ComfyUI/commit/

.github/workflows/docker.yml

@@ -1,23 +0,0 @@
-name: Build Images
-
-on:
-  push:
-    branches: master
-  pull_request:
-      paths:
-        - docker-compose.yml
-        - services
-
-jobs:
-  build:
-    strategy:
-      matrix:
-        profile:
-          - auto
-          - comfy
-          - download
-    runs-on: ubuntu-latest
-    name: ${{ matrix.profile }}
-    steps:
-      - uses: actions/checkout@v3
-      - run: docker compose --profile ${{ matrix.profile }} build --progress plain

.github/workflows/stale.yml

@@ -1,20 +0,0 @@
-name: 'Close stale issues and PRs'
-on:
-  schedule:
-    - cron: '0 0 * * *'
-
-jobs:
-  stale:
-    runs-on: ubuntu-latest
-    steps:
-      - uses: actions/stale@v6
-        with:
-          only-labels: awaiting-response
-          stale-issue-message: This issue is stale because it has been open 14 days with no activity. Remove stale label or comment or this will be closed in 7 days.
-          stale-pr-message: This PR is stale because it has been open 14 days with no activity. Remove stale label or comment or this will be closed in 7 days.
-          close-issue-message: This issue was closed because it has been stalled for 7 days with no activity.
-          close-pr-message: This PR was closed because it has been stalled for 7 days with no activity.
-          days-before-issue-stale: 14
-          days-before-pr-stale: 14
-          days-before-issue-close: 7
-          days-before-pr-close: 7

.gitignore

@@ -1,6 +1,2 @@
+/dev
 /.devcontainer
-/docker-compose.override.yml
-
-# VSCode specific
-*.code-workspace
-/.vscode

.vscode/launch.json

@@ -1,22 +0,0 @@
-{
-  // Use IntelliSense to learn about possible attributes.
-  // Hover to view descriptions of existing attributes.
-  // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
-  "version": "0.2.0",
-  "configurations": [
-    {
-      "name": "WebUI",
-      "type": "python",
-      "request": "launch",
-      "program": "${file}",
-      "cwd": "/stable-diffusion",
-      "args": ["--ckpt", "${workspaceFolder}/models/model.ckpt", "--gfpgan-dir", "${workspaceFolder}/models/", "--extra-models-cpu"],
-      "env": {
-        "TRANSFORMERS_CACHE":"${workspaceFolder}/cache/transformers",
-        "TORCH_HOME":"${workspaceFolder}/cache/torch"
-      },
-      "console": "integratedTerminal",
-      "justMyCode": false
-    }
-  ]
-}

LICENSE

@@ -86,4 +86,4 @@ administration of justice, law enforcement, immigration or asylum
 processes, such as predicting an individual will commit fraud/crime
 commitment (e.g. by text profiling, drawing causal relationships between
 assertions made in documents, indiscriminate and arbitrarily-targeted
-use).
+use).

build/Dockerfile

@@ -0,0 +1,31 @@
+# syntax=docker/dockerfile:1
+
+FROM continuumio/miniconda3:4.12.0
+
+
+RUN conda install python=3.8.5 && conda clean -a -y
+RUN conda install pytorch==1.11.0 torchvision==0.12.0 cudatoolkit=11.3 -c pytorch && conda clean -a -y
+RUN git clone https://github.com/hlky/stable-diffusion.git && cd stable-diffusion && git reset --hard 554bd068e6f2f6bc55449a67fe017ddd77090f28
+RUN conda env update --file stable-diffusion/environment.yaml --name base && conda clean -a -y
+
+# fonts
+RUN apt-get update && apt install fonts-dejavu-core && apt-get clean
+
+RUN git clone https://github.com/hlky/stable-diffusion-webui.git && cd stable-diffusion-webui && \
+  git reset --hard 33890563d237889aca7cbc4f04ae42b885aee07e && \
+  cp -t /stable-diffusion/scripts/ txt2img.yaml webui.py webui.yaml webui_playground.py && \
+  cd / && rm -rf stable-diffusion-webui
+
+
+WORKDIR /stable-diffusion
+ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS=""
+
+EXPOSE 7860
+CMD ln -sf /models/model.ckpt /stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt && \
+  ln -sf /models/GFPGANv1.3.pth   /stable-diffusion/src/gfpgan/experiments/pretrained_models/GFPGANv1.3.pth && \
+  ln -sf /models/RealESRGAN_x4plus.pth   /stable-diffusion/src/realesrgan/experiments/pretrained_models/RealESRGAN_x4plus.pth && \
+  # force facexlib cache
+  mkdir -p /cache/weights/ && rm -rf /opt/conda/lib/python3.8/site-packages/facexlib/weights && \
+  ln -sf  /cache/weights/ /opt/conda/lib/python3.8/site-packages/facexlib/ && \
+  # run, -u to not buffer stdout / stderr
+  python3 -u scripts/webui.py --outdir /output --save-metadata --ckpt /models/model.ckpt ${CLI_ARGS}

cache/.gitignore

@@ -0,0 +1,3 @@
+/torch
+/transformers
+/weights

data/.gitignore

@@ -1,4 +0,0 @@
-/.cache
-/config
-/embeddings
-/models

docker-compose.yml

@@ -1,55 +1,21 @@
-x-base_service: &base_service
+version: '3.9'
+
+services:
+  model:
+    build: ./build/
+    restart: on-failure
     ports:
-      - "${WEBUI_PORT:-7860}:7860"
+      - "7860:7860"
     volumes:
-      - &v1 ./data:/data
-      - &v2 ./output:/output
-    stop_signal: SIGKILL
-    tty: true
+      - ./cache:/cache
+      - ./output:/output
+      - ./models:/models
+    environment:
+      - CLI_ARGS=--extra-models-cpu
     deploy:
       resources:
         reservations:
           devices:
               - driver: nvidia
                 device_ids: ['0']
-                capabilities: [compute, utility]
-
-name: webui-docker
-
-services:
-  download:
-    build: ./services/download/
-    profiles: ["download"]
-    volumes:
-      - *v1
-
-  auto: &automatic
-    <<: *base_service
-    profiles: ["auto"]
-    build: ./services/AUTOMATIC1111
-    image: sd-auto:78
-    environment:
-      - CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
-
-  auto-cpu:
-    <<: *automatic
-    profiles: ["auto-cpu"]
-    deploy: {}
-    environment:
-      - CLI_ARGS=--no-half --precision full --allow-code --enable-insecure-extension-access --api
-
-  comfy: &comfy
-    <<: *base_service
-    profiles: ["comfy"]
-    build: ./services/comfy/
-    image: sd-comfy:7
-    environment:
-      - CLI_ARGS=
-
-
-  comfy-cpu:
-    <<: *comfy
-    profiles: ["comfy-cpu"]
-    deploy: {}
-    environment:
-      - CLI_ARGS=--cpu
+                capabilities: [gpu]

models/.gitignore

@@ -0,0 +1,4 @@
+/model.ckpt
+/GFPGANv1.3.pth
+/RealESRGAN_x4plus.pth
+/RealESRGAN_x4plus_anime_6B.pth

services/AUTOMATIC1111/Dockerfile

@@ -1,66 +0,0 @@
-FROM alpine/git:2.36.2 as download
-
-COPY clone.sh /clone.sh
-
-RUN . /clone.sh stable-diffusion-webui-assets https://github.com/AUTOMATIC1111/stable-diffusion-webui-assets.git 6f7db241d2f8ba7457bac5ca9753331f0c266917
-
-RUN . /clone.sh stable-diffusion-stability-ai https://github.com/Stability-AI/stablediffusion.git cf1d67a6fd5ea1aa600c4df58e5b47da45f6bdbf \
-  && rm -rf assets data/**/*.png data/**/*.jpg data/**/*.gif
-
-RUN . /clone.sh BLIP https://github.com/salesforce/BLIP.git 48211a1594f1321b00f14c9f7a5b4813144b2fb9
-RUN . /clone.sh k-diffusion https://github.com/crowsonkb/k-diffusion.git ab527a9a6d347f364e3d185ba6d714e22d80cb3c
-RUN . /clone.sh clip-interrogator https://github.com/pharmapsychotic/clip-interrogator 2cf03aaf6e704197fd0dae7c7f96aa59cf1b11c9
-RUN . /clone.sh generative-models https://github.com/Stability-AI/generative-models 45c443b316737a4ab6e40413d7794a7f5657c19f
-RUN . /clone.sh stable-diffusion-webui-assets https://github.com/AUTOMATIC1111/stable-diffusion-webui-assets 6f7db241d2f8ba7457bac5ca9753331f0c266917
-
-
-FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
-
-ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
-
-RUN --mount=type=cache,target=/var/cache/apt \
-  apt-get update && \
-  # we need those
-  apt-get install -y fonts-dejavu-core rsync git jq moreutils aria2 \
-  # extensions needs those
-  ffmpeg libglfw3-dev libgles2-mesa-dev pkg-config libcairo2 libcairo2-dev build-essential
-
-
-WORKDIR /
-RUN --mount=type=cache,target=/root/.cache/pip \
-  git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git && \
-  cd stable-diffusion-webui && \
-  git reset --hard v1.9.4 && \
-  pip install -r requirements_versions.txt
-
-
-ENV ROOT=/stable-diffusion-webui
-
-COPY --from=download /repositories/ ${ROOT}/repositories/
-RUN mkdir ${ROOT}/interrogate && cp ${ROOT}/repositories/clip-interrogator/clip_interrogator/data/* ${ROOT}/interrogate
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-  pip install pyngrok xformers==0.0.26.post1 \
-  git+https://github.com/TencentARC/GFPGAN.git@8d2447a2d918f8eba5a4a01463fd48e45126a379 \
-  git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1 \
-  git+https://github.com/mlfoundations/open_clip.git@v2.20.0
-
-# there seems to be a memory leak (or maybe just memory not being freed fast enough) that is fixed by this version of malloc
-# maybe move this up to the dependencies list.
-RUN apt-get -y install libgoogle-perftools-dev && apt-get clean
-ENV LD_PRELOAD=libtcmalloc.so
-
-COPY . /docker
-
-RUN \
-  # mv ${ROOT}/style.css ${ROOT}/user.css && \
-  # one of the ugliest hacks I ever wrote \
-  sed -i 's/in_app_dir = .*/in_app_dir = True/g' /opt/conda/lib/python3.10/site-packages/gradio/routes.py && \
-  git config --global --add safe.directory '*'
-
-WORKDIR ${ROOT}
-ENV NVIDIA_VISIBLE_DEVICES=all
-ENV CLI_ARGS=""
-EXPOSE 7860
-ENTRYPOINT ["/docker/entrypoint.sh"]
-CMD python -u webui.py --listen --port 7860 ${CLI_ARGS}

services/AUTOMATIC1111/clone.sh

@@ -1,11 +0,0 @@
-#!/bin/bash
-
-set -Eeuox pipefail
-
-mkdir -p /repositories/"$1"
-cd /repositories/"$1"
-git init
-git remote add origin "$2"
-git fetch origin "$3" --depth=1
-git reset --hard "$3"
-rm -rf .git

services/AUTOMATIC1111/config.py

@@ -1,78 +0,0 @@
-#!/usr/bin/env python3
-
-"""Checks and sets default values for config.json before starting the container."""
-
-import json
-import re
-import os.path
-import sys
-
-DEFAULT_FILEPATH = '/data/config/auto/config.json'
-
-DEFAULT_OUTDIRS = {
-  "outdir_samples": "",
-  "outdir_txt2img_samples": "/output/txt2img",
-  "outdir_img2img_samples": "/output/img2img",
-  "outdir_extras_samples": "/output/extras",
-  "outdir_grids": "",
-  "outdir_txt2img_grids": "/output/txt2img-grids",
-  "outdir_img2img_grids": "/output/img2img-grids",
-  "outdir_save": "/output/saved",
-  "outdir_init_images": "/output/init-images",
-}
-RE_VALID_OUTDIR = re.compile(r"(^/output(/\.?[\w\-\_]+)+/?$)|(^\s?$)")
-
-DEFAULT_OTHER = {
-  "font": "DejaVuSans.ttf",
-}
-
-def dict_to_json_file(target_file: str, data: dict):
-  """Write dictionary to specified json file"""
-
-  with open(target_file, 'w') as f:
-    json.dump(data, f)
-
-def json_file_to_dict(config_file: str) -> dict|None:
-   """Load json file into a dictionary. Return None if file does not exist."""
-
-   if os.path.isfile(config_file):
-    with open(config_file, 'r') as f:
-      return json.load(f)
-   else:
-      return None
-
-def replace_if_invalid(value: str, replacement: str, pattern: str|re.Pattern[str]) -> str:
-  """Returns original value if valid, fallback value if invalid"""
-
-  if re.match(pattern, value):
-    return value
-  else:
-    return replacement
-
-def check_and_replace_config(config_file: str, target_file: str = None):
-  """Checks given file for invalid values. Replaces those with fallback values (default: overwrites file)."""
-
-  # Get current user config, or empty if file does not exists
-  data = json_file_to_dict(config_file) or {}
-
-  # Check and fix output directories
-  for k, def_val in DEFAULT_OUTDIRS.items():
-    if k not in data:
-      data[k] = def_val
-    else:
-      data[k] = replace_if_invalid(value=data[k], replacement=def_val, pattern=RE_VALID_OUTDIR)
-
-  # Check and fix other default settings
-  for k, def_val in DEFAULT_OTHER.items():
-    if k not in data:
-      data[k] = def_val
-
-  # Write results to file
-  dict_to_json_file(target_file or config_file, data)
-
-if __name__ == '__main__':
-  if len(sys.argv) > 1:
-    check_and_replace_config(*sys.argv[1:])
-  else:
-    check_and_replace_config(DEFAULT_FILEPATH)
-

services/AUTOMATIC1111/entrypoint.sh

@@ -1,85 +0,0 @@
-#!/bin/bash
-
-set -Eeuo pipefail
-
-# TODO: move all mkdir -p ?
-mkdir -p /data/config/auto/scripts/
-# mount scripts individually
-
-echo $ROOT
-ls -lha $ROOT
-
-find "${ROOT}/scripts/" -maxdepth 1 -type l -delete
-cp -vrfTs /data/config/auto/scripts/ "${ROOT}/scripts/"
-
-# Set up config file
-python /docker/config.py /data/config/auto/config.json
-
-if [ ! -f /data/config/auto/ui-config.json ]; then
-  echo '{}' >/data/config/auto/ui-config.json
-fi
-
-if [ ! -f /data/config/auto/styles.csv ]; then
-  touch /data/config/auto/styles.csv
-fi
-
-# copy models from original models folder
-mkdir -p /data/models/VAE-approx/ /data/models/karlo/
-
-rsync -a --info=NAME ${ROOT}/models/VAE-approx/ /data/models/VAE-approx/
-rsync -a --info=NAME ${ROOT}/models/karlo/ /data/models/karlo/
-
-declare -A MOUNTS
-
-MOUNTS["/root/.cache"]="/data/.cache"
-MOUNTS["${ROOT}/models"]="/data/models"
-
-MOUNTS["${ROOT}/embeddings"]="/data/embeddings"
-MOUNTS["${ROOT}/config.json"]="/data/config/auto/config.json"
-MOUNTS["${ROOT}/ui-config.json"]="/data/config/auto/ui-config.json"
-MOUNTS["${ROOT}/styles.csv"]="/data/config/auto/styles.csv"
-MOUNTS["${ROOT}/extensions"]="/data/config/auto/extensions"
-MOUNTS["${ROOT}/config_states"]="/data/config/auto/config_states"
-
-# extra hacks
-MOUNTS["${ROOT}/repositories/CodeFormer/weights/facelib"]="/data/.cache"
-
-for to_path in "${!MOUNTS[@]}"; do
-  set -Eeuo pipefail
-  from_path="${MOUNTS[${to_path}]}"
-  rm -rf "${to_path}"
-  if [ ! -f "$from_path" ]; then
-    mkdir -vp "$from_path"
-  fi
-  mkdir -vp "$(dirname "${to_path}")"
-  ln -sT "${from_path}" "${to_path}"
-  echo Mounted $(basename "${from_path}")
-done
-
-echo "Installing extension dependencies (if any)"
-
-# because we build our container as root:
-chown -R root ~/.cache/
-chmod 766 ~/.cache/
-
-shopt -s nullglob
-# For install.py, please refer to https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Developing-extensions#installpy
-list=(./extensions/*/install.py)
-for installscript in "${list[@]}"; do
-  EXTNAME=$(echo $installscript | cut -d '/' -f 3)
-  # Skip installing dependencies if extension is disabled in config
-  if $(jq -e ".disabled_extensions|any(. == \"$EXTNAME\")" config.json); then
-    echo "Skipping disabled extension ($EXTNAME)"
-    continue
-  fi
-  PYTHONPATH=${ROOT} python "$installscript"
-done
-
-if [ -f "/data/config/auto/startup.sh" ]; then
-  pushd ${ROOT}
-  echo "Running startup script"
-  . /data/config/auto/startup.sh
-  popd
-fi
-
-exec "$@"

services/comfy/Dockerfile

@@ -1,22 +0,0 @@
-FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
-
-ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
-
-RUN apt-get update && apt-get install -y git && apt-get clean
-
-ENV ROOT=/stable-diffusion
-RUN --mount=type=cache,target=/root/.cache/pip \
-  git clone https://github.com/comfyanonymous/ComfyUI.git ${ROOT} && \
-  cd ${ROOT} && \
-  git checkout master && \
-  git reset --hard 276f8fce9f5a80b500947fb5745a4dde9e84622d && \
-  pip install -r requirements.txt
-
-WORKDIR ${ROOT}
-COPY . /docker/
-RUN chmod u+x /docker/entrypoint.sh && cp /docker/extra_model_paths.yaml ${ROOT}
-
-ENV NVIDIA_VISIBLE_DEVICES=all PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
-EXPOSE 7860
-ENTRYPOINT ["/docker/entrypoint.sh"]
-CMD python -u main.py --listen --port 7860 ${CLI_ARGS}

services/comfy/entrypoint.sh

@@ -1,31 +0,0 @@
-#!/bin/bash
-
-set -Eeuo pipefail
-
-mkdir -vp /data/config/comfy/custom_nodes
-
-declare -A MOUNTS
-
-MOUNTS["/root/.cache"]="/data/.cache"
-MOUNTS["${ROOT}/input"]="/data/config/comfy/input"
-MOUNTS["${ROOT}/output"]="/output/comfy"
-
-for to_path in "${!MOUNTS[@]}"; do
-  set -Eeuo pipefail
-  from_path="${MOUNTS[${to_path}]}"
-  rm -rf "${to_path}"
-  if [ ! -f "$from_path" ]; then
-    mkdir -vp "$from_path"
-  fi
-  mkdir -vp "$(dirname "${to_path}")"
-  ln -sT "${from_path}" "${to_path}"
-  echo Mounted $(basename "${from_path}")
-done
-
-if [ -f "/data/config/comfy/startup.sh" ]; then
-  pushd ${ROOT}
-  . /data/config/comfy/startup.sh
-  popd
-fi
-
-exec "$@"

services/comfy/extra_model_paths.yaml

@@ -1,25 +0,0 @@
-a111:
-  base_path: /data
-
-  checkpoints: models/Stable-diffusion
-  configs: models/Stable-diffusion
-  vae: models/VAE
-  loras: models/Lora
-  upscale_models: |
-    models/RealESRGAN
-    models/ESRGAN
-    models/SwinIR
-    models/GFPGAN
-  hypernetworks: models/hypernetworks
-  controlnet: models/ControlNet
-  gligen: models/GLIGEN
-  clip: models/CLIPEncoder
-  embeddings: embeddings
-
-  custom_nodes: config/comfy/custom_nodes
-
-  # TODO: I am unsure about these, need more testing
-  # style_models: config/comfy/style_models
-  # t2i_adapter: config/comfy/t2i_adapter
-  # clip_vision: config/comfy/clip_vision
-  # diffusers: config/comfy/diffusers

services/download/Dockerfile

@@ -1,6 +0,0 @@
-FROM bash:alpine3.19
-
-RUN apk update && apk add parallel aria2
-COPY . /docker
-RUN chmod +x /docker/download.sh
-ENTRYPOINT ["/docker/download.sh"]

services/download/checksums.sha256

@@ -1,8 +0,0 @@
-cc6cb27103417325ff94f52b7a5d2dde45a7515b25c255d8e396c90014281516  /data/models/Stable-diffusion/v1-5-pruned-emaonly.ckpt
-c6bbc15e3224e6973459ba78de4998b80b50112b0ae5b5c67113d56b4e366b19  /data/models/Stable-diffusion/sd-v1-5-inpainting.ckpt
-c6a580b13a5bc05a5e16e4dbb80608ff2ec251a162311590c1f34c013d7f3dab  /data/models/VAE/vae-ft-mse-840000-ema-pruned.ckpt
-e2cd4703ab14f4d01fd1383a8a8b266f9a5833dacee8e6a79d3bf21a1b6be5ad  /data/models/GFPGAN/GFPGANv1.4.pth
-4fa0d38905f75ac06eb49a7951b426670021be3018265fd191d2125df9d682f1  /data/models/RealESRGAN/RealESRGAN_x4plus.pth
-f872d837d3c90ed2e05227bed711af5671a6fd1c9f7d7e91c911a61f155e99da  /data/models/RealESRGAN/RealESRGAN_x4plus_anime_6B.pth
-c209caecac2f97b4bb8f4d726b70ac2ac9b35904b7fc99801e1f5e61f9210c13  /data/models/LDSR/model.ckpt
-9d6ad53c5dafeb07200fb712db14b813b527edd262bc80ea136777bdb41be2ba  /data/models/LDSR/project.yaml

services/download/download.sh

@@ -1,31 +0,0 @@
-#!/usr/bin/env bash
-
-set -Eeuo pipefail
-
-# TODO: maybe just use the .gitignore file to create all of these
-mkdir -vp /data/.cache \
-  /data/embeddings \
-  /data/config/ \
-  /data/models/ \
-  /data/models/Stable-diffusion \
-  /data/models/GFPGAN \
-  /data/models/RealESRGAN \
-  /data/models/LDSR \
-  /data/models/VAE
-
-echo "Downloading, this might take a while..."
-
-aria2c -x 10 --disable-ipv6 --input-file /docker/links.txt --dir /data/models --continue
-
-echo "Checking SHAs..."
-
-parallel --will-cite -a /docker/checksums.sha256 "echo -n {} | sha256sum -c"
-
-cat <<EOF
-By using this software, you agree to the following licenses:
-https://github.com/AbdBarho/stable-diffusion-webui-docker/blob/master/LICENSE
-https://github.com/CompVis/stable-diffusion/blob/main/LICENSE
-https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/master/LICENSE.txt
-https://github.com/invoke-ai/InvokeAI/blob/main/LICENSE
-And licenses of all UIs, third party libraries, and extensions.
-EOF

services/download/links.txt

@@ -1,16 +0,0 @@
-https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt
-  out=Stable-diffusion/v1-5-pruned-emaonly.ckpt
-https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.ckpt
-  out=VAE/vae-ft-mse-840000-ema-pruned.ckpt
-https://huggingface.co/runwayml/stable-diffusion-inpainting/resolve/main/sd-v1-5-inpainting.ckpt
-  out=Stable-diffusion/sd-v1-5-inpainting.ckpt
-https://github.com/TencentARC/GFPGAN/releases/download/v1.3.4/GFPGANv1.4.pth
-  out=GFPGAN/GFPGANv1.4.pth
-https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth
-  out=RealESRGAN/RealESRGAN_x4plus.pth
-https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth
-  out=RealESRGAN/RealESRGAN_x4plus_anime_6B.pth
-https://heibox.uni-heidelberg.de/f/31a76b13ea27482981b4/?dl=1
-  out=LDSR/project.yaml
-https://heibox.uni-heidelberg.de/f/578df07c8fc04ffbadf3/?dl=1
-  out=LDSR/model.ckpt