mirror of
https://github.com/AbdBarho/stable-diffusion-webui-docker.git
synced 2025-10-27 08:14:26 -04:00
Compare commits
11 Commits
8.2.0
...
fcf5486879
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
fcf5486879 | ||
|
|
cae1586b45 | ||
|
|
802d0bcd68 | ||
|
|
b1a26b8041 | ||
|
|
f1bf3b0943 | ||
|
|
35a18b3d46 | ||
|
|
887e49c495 | ||
|
|
7051ce0a44 | ||
|
|
ac94eac2b5 | ||
|
|
015c2ec829 | ||
|
|
245d1d443f |
1
.github/workflows/docker.yml
vendored
1
.github/workflows/docker.yml
vendored
@@ -14,7 +14,6 @@ jobs:
|
|||||||
matrix:
|
matrix:
|
||||||
profile:
|
profile:
|
||||||
- auto
|
- auto
|
||||||
- invoke
|
|
||||||
- comfy
|
- comfy
|
||||||
- download
|
- download
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
|
|||||||
@@ -18,14 +18,6 @@ This repository provides multiple UIs for you to play around with stable diffusi
|
|||||||
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
||||||
|  |  |  |
|
|  |  |  |
|
||||||
|
|
||||||
### [InvokeAI](https://github.com/invoke-ai/InvokeAI)
|
|
||||||
|
|
||||||
[Full feature list here](https://github.com/invoke-ai/InvokeAI#features), Screenshots:
|
|
||||||
|
|
||||||
| Text to image | Image to image | Extras |
|
|
||||||
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
|
||||||
|  |  |  |
|
|
||||||
|
|
||||||
### [ComfyUI](https://github.com/comfyanonymous/ComfyUI)
|
### [ComfyUI](https://github.com/comfyanonymous/ComfyUI)
|
||||||
|
|
||||||
[Full feature list here](https://github.com/comfyanonymous/ComfyUI#features), Screenshot:
|
[Full feature list here](https://github.com/comfyanonymous/ComfyUI#features), Screenshot:
|
||||||
|
|||||||
58
docker-compose.yml
Normal file → Executable file
58
docker-compose.yml
Normal file → Executable file
@@ -1,20 +1,18 @@
|
|||||||
version: '3.9'
|
|
||||||
|
|
||||||
x-base_service: &base_service
|
x-base_service: &base_service
|
||||||
ports:
|
ports:
|
||||||
- "${WEBUI_PORT:-7860}:7860"
|
- "${WEBUI_PORT:-7860}:7860"
|
||||||
volumes:
|
volumes:
|
||||||
- &v1 ./data:/data
|
- &v1 ./data:/data
|
||||||
- &v2 ./output:/output
|
- &v2 ./output:/output
|
||||||
stop_signal: SIGKILL
|
stop_signal: SIGKILL
|
||||||
tty: true
|
tty: true
|
||||||
deploy:
|
deploy:
|
||||||
resources:
|
resources:
|
||||||
reservations:
|
reservations:
|
||||||
devices:
|
devices:
|
||||||
- driver: nvidia
|
- driver: nvidia
|
||||||
device_ids: ['0']
|
device_ids: ["0"]
|
||||||
capabilities: [compute, utility]
|
capabilities: [compute, utility]
|
||||||
|
|
||||||
name: webui-docker
|
name: webui-docker
|
||||||
|
|
||||||
@@ -29,7 +27,7 @@ services:
|
|||||||
<<: *base_service
|
<<: *base_service
|
||||||
profiles: ["auto"]
|
profiles: ["auto"]
|
||||||
build: ./services/AUTOMATIC1111
|
build: ./services/AUTOMATIC1111
|
||||||
image: sd-auto:71
|
image: sd-auto:78
|
||||||
environment:
|
environment:
|
||||||
- CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
|
- CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
|
||||||
|
|
||||||
@@ -40,31 +38,29 @@ services:
|
|||||||
environment:
|
environment:
|
||||||
- CLI_ARGS=--no-half --precision full --allow-code --enable-insecure-extension-access --api
|
- CLI_ARGS=--no-half --precision full --allow-code --enable-insecure-extension-access --api
|
||||||
|
|
||||||
invoke: &invoke
|
forge: &forge
|
||||||
<<: *base_service
|
<<: *base_service
|
||||||
profiles: ["invoke"]
|
profiles: ["forge"]
|
||||||
build: ./services/invoke/
|
build: ./services/AUTOMATIC1111-forge
|
||||||
image: sd-invoke:30
|
image: sd-auto-forge:78
|
||||||
environment:
|
environment:
|
||||||
- PRELOAD=true
|
- CLI_ARGS=--allow-code --xformers --enable-insecure-extension-access --api
|
||||||
- CLI_ARGS=--xformers
|
|
||||||
|
|
||||||
# invoke-cpu:
|
forge-cpu:
|
||||||
# <<: *invoke
|
<<: *forge
|
||||||
# profiles: ["invoke-cpu"]
|
profiles: ["forge-cpu"]
|
||||||
# environment:
|
deploy: {}
|
||||||
# - PRELOAD=true
|
environment:
|
||||||
# - CLI_ARGS=--always_use_cpu
|
- CLI_ARGS=--no-half --precision full --allow-code --enable-insecure-extension-access --api
|
||||||
|
|
||||||
comfy: &comfy
|
comfy: &comfy
|
||||||
<<: *base_service
|
<<: *base_service
|
||||||
profiles: ["comfy"]
|
profiles: ["comfy"]
|
||||||
build: ./services/comfy/
|
build: ./services/comfy/
|
||||||
image: sd-comfy:6
|
image: sd-comfy:7
|
||||||
environment:
|
environment:
|
||||||
- CLI_ARGS=
|
- CLI_ARGS=
|
||||||
|
|
||||||
|
|
||||||
comfy-cpu:
|
comfy-cpu:
|
||||||
<<: *comfy
|
<<: *comfy
|
||||||
profiles: ["comfy-cpu"]
|
profiles: ["comfy-cpu"]
|
||||||
|
|||||||
68
services/AUTOMATIC1111-forge/Dockerfile
Executable file
68
services/AUTOMATIC1111-forge/Dockerfile
Executable file
@@ -0,0 +1,68 @@
|
|||||||
|
FROM alpine/git:latest 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 huggingface_guess https://github.com/lllyasviel/huggingface_guess.git 70942022b6bcd17d941c1b4172804175758618e2
|
||||||
|
RUN . /clone.sh google_blockly_prototypes https://github.com/lllyasviel/google_blockly_prototypes.git 1e98997c7fedaf5106af9849b6f50ebe5c4408f1
|
||||||
|
RUN . /clone.sh stable-diffusion-webui-assets https://github.com/AUTOMATIC1111/stable-diffusion-webui-assets.git 6f7db241d2f8ba7457bac5ca9753331f0c266917
|
||||||
|
|
||||||
|
|
||||||
|
FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-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
|
||||||
|
|
||||||
|
|
||||||
|
ENV ROOT=/stable-diffusion-webui-forge
|
||||||
|
|
||||||
|
WORKDIR /
|
||||||
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
||||||
|
git clone https://github.com/lllyasviel/stable-diffusion-webui-forge.git && \
|
||||||
|
cd $ROOT && \
|
||||||
|
pip install -r requirements_versions.txt
|
||||||
|
|
||||||
|
RUN pip install -U typing_extensions
|
||||||
|
|
||||||
|
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.27 \
|
||||||
|
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}
|
||||||
15
services/AUTOMATIC1111-forge/clone.sh
Executable file
15
services/AUTOMATIC1111-forge/clone.sh
Executable file
@@ -0,0 +1,15 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
set -Eeox pipefail
|
||||||
|
|
||||||
|
mkdir -p /repositories/"$1"
|
||||||
|
cd /repositories/"$1"
|
||||||
|
git init
|
||||||
|
git remote add origin "$2"
|
||||||
|
|
||||||
|
if [ -n "$3" ]; then
|
||||||
|
git fetch origin "$3" --depth=1
|
||||||
|
git reset --hard "$3"
|
||||||
|
fi
|
||||||
|
|
||||||
|
rm -rf .git
|
||||||
78
services/AUTOMATIC1111-forge/config.py
Executable file
78
services/AUTOMATIC1111-forge/config.py
Executable file
@@ -0,0 +1,78 @@
|
|||||||
|
#!/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/forge/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)
|
||||||
|
|
||||||
85
services/AUTOMATIC1111-forge/entrypoint.sh
Executable file
85
services/AUTOMATIC1111-forge/entrypoint.sh
Executable file
@@ -0,0 +1,85 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
set -Eeuo pipefail
|
||||||
|
|
||||||
|
# TODO: move all mkdir -p ?
|
||||||
|
mkdir -p /data/config/forge/scripts/
|
||||||
|
# mount scripts individually
|
||||||
|
|
||||||
|
echo $ROOT
|
||||||
|
ls -lha $ROOT
|
||||||
|
|
||||||
|
find "${ROOT}/scripts/" -maxdepth 1 -type l -delete
|
||||||
|
cp -vrfTs /data/config/forge/scripts/ "${ROOT}/scripts/"
|
||||||
|
|
||||||
|
# Set up config file
|
||||||
|
python /docker/config.py /data/config/forge/config.json
|
||||||
|
|
||||||
|
if [ ! -f /data/config/forge/ui-config.json ]; then
|
||||||
|
echo '{}' >/data/config/forge/ui-config.json
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ ! -f /data/config/forge/styles.csv ]; then
|
||||||
|
touch /data/config/forge/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/forge/config.json"
|
||||||
|
MOUNTS["${ROOT}/ui-config.json"]="/data/config/forge/ui-config.json"
|
||||||
|
MOUNTS["${ROOT}/styles.csv"]="/data/config/forge/styles.csv"
|
||||||
|
MOUNTS["${ROOT}/extensions"]="/data/config/forge/extensions"
|
||||||
|
MOUNTS["${ROOT}/config_states"]="/data/config/forge/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/forge/startup.sh" ]; then
|
||||||
|
pushd ${ROOT}
|
||||||
|
echo "Running startup script"
|
||||||
|
. /data/config/forge/startup.sh
|
||||||
|
popd
|
||||||
|
fi
|
||||||
|
|
||||||
|
exec "$@"
|
||||||
16
services/AUTOMATIC1111/Dockerfile
Normal file → Executable file
16
services/AUTOMATIC1111/Dockerfile
Normal file → Executable file
@@ -2,20 +2,19 @@ FROM alpine/git:2.36.2 as download
|
|||||||
|
|
||||||
COPY clone.sh /clone.sh
|
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 \
|
RUN . /clone.sh stable-diffusion-stability-ai https://github.com/Stability-AI/stablediffusion.git cf1d67a6fd5ea1aa600c4df58e5b47da45f6bdbf \
|
||||||
&& rm -rf assets data/**/*.png data/**/*.jpg data/**/*.gif
|
&& rm -rf assets data/**/*.png data/**/*.jpg data/**/*.gif
|
||||||
|
|
||||||
RUN . /clone.sh CodeFormer https://github.com/sczhou/CodeFormer.git c5b4593074ba6214284d6acd5f1719b6c5d739af \
|
|
||||||
&& rm -rf assets inputs
|
|
||||||
|
|
||||||
RUN . /clone.sh BLIP https://github.com/salesforce/BLIP.git 48211a1594f1321b00f14c9f7a5b4813144b2fb9
|
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 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 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 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.1.2-cuda12.1-cudnn8-runtime
|
FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
|
||||||
|
|
||||||
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
|
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
|
||||||
|
|
||||||
@@ -31,24 +30,23 @@ WORKDIR /
|
|||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
||||||
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git && \
|
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git && \
|
||||||
cd stable-diffusion-webui && \
|
cd stable-diffusion-webui && \
|
||||||
git reset --hard cf2772fab0af5573da775e7437e6acdca424f26e && \
|
git reset --hard v1.10.1 && \
|
||||||
pip install -r requirements_versions.txt
|
pip install -r requirements_versions.txt
|
||||||
|
|
||||||
|
RUN pip install -U typing_extensions
|
||||||
|
|
||||||
ENV ROOT=/stable-diffusion-webui
|
ENV ROOT=/stable-diffusion-webui
|
||||||
|
|
||||||
COPY --from=download /repositories/ ${ROOT}/repositories/
|
COPY --from=download /repositories/ ${ROOT}/repositories/
|
||||||
RUN mkdir ${ROOT}/interrogate && cp ${ROOT}/repositories/clip-interrogator/clip_interrogator/data/* ${ROOT}/interrogate
|
RUN mkdir ${ROOT}/interrogate && cp ${ROOT}/repositories/clip-interrogator/clip_interrogator/data/* ${ROOT}/interrogate
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
||||||
pip install -r ${ROOT}/repositories/CodeFormer/requirements.txt
|
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
||||||
pip install pyngrok xformers \
|
pip install pyngrok xformers==0.0.26.post1 \
|
||||||
git+https://github.com/TencentARC/GFPGAN.git@8d2447a2d918f8eba5a4a01463fd48e45126a379 \
|
git+https://github.com/TencentARC/GFPGAN.git@8d2447a2d918f8eba5a4a01463fd48e45126a379 \
|
||||||
git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1 \
|
git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1 \
|
||||||
git+https://github.com/mlfoundations/open_clip.git@v2.20.0
|
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 eno8ugh) that is fixed by this version of malloc
|
# 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.
|
# maybe move this up to the dependencies list.
|
||||||
RUN apt-get -y install libgoogle-perftools-dev && apt-get clean
|
RUN apt-get -y install libgoogle-perftools-dev && apt-get clean
|
||||||
ENV LD_PRELOAD=libtcmalloc.so
|
ENV LD_PRELOAD=libtcmalloc.so
|
||||||
|
|||||||
10
services/AUTOMATIC1111/clone.sh
Normal file → Executable file
10
services/AUTOMATIC1111/clone.sh
Normal file → Executable file
@@ -1,11 +1,15 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
set -Eeuox pipefail
|
set -Eeox pipefail
|
||||||
|
|
||||||
mkdir -p /repositories/"$1"
|
mkdir -p /repositories/"$1"
|
||||||
cd /repositories/"$1"
|
cd /repositories/"$1"
|
||||||
git init
|
git init
|
||||||
git remote add origin "$2"
|
git remote add origin "$2"
|
||||||
git fetch origin "$3" --depth=1
|
|
||||||
git reset --hard "$3"
|
if [ -n "$3" ]; then
|
||||||
|
git fetch origin "$3" --depth=1
|
||||||
|
git reset --hard "$3"
|
||||||
|
fi
|
||||||
|
|
||||||
rm -rf .git
|
rm -rf .git
|
||||||
|
|||||||
0
services/AUTOMATIC1111/config.py
Normal file → Executable file
0
services/AUTOMATIC1111/config.py
Normal file → Executable file
3
services/comfy/Dockerfile
Normal file → Executable file
3
services/comfy/Dockerfile
Normal file → Executable file
@@ -1,4 +1,4 @@
|
|||||||
FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime
|
FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime
|
||||||
|
|
||||||
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
|
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
|
||||||
|
|
||||||
@@ -9,7 +9,6 @@ RUN --mount=type=cache,target=/root/.cache/pip \
|
|||||||
git clone https://github.com/comfyanonymous/ComfyUI.git ${ROOT} && \
|
git clone https://github.com/comfyanonymous/ComfyUI.git ${ROOT} && \
|
||||||
cd ${ROOT} && \
|
cd ${ROOT} && \
|
||||||
git checkout master && \
|
git checkout master && \
|
||||||
git reset --hard d1f3637a5a944d0607b899babd8ff11d87100503 && \
|
|
||||||
pip install -r requirements.txt
|
pip install -r requirements.txt
|
||||||
|
|
||||||
WORKDIR ${ROOT}
|
WORKDIR ${ROOT}
|
||||||
|
|||||||
0
services/comfy/extra_model_paths.yaml
Normal file → Executable file
0
services/comfy/extra_model_paths.yaml
Normal file → Executable file
4
services/download/Dockerfile
Normal file → Executable file
4
services/download/Dockerfile
Normal file → Executable file
@@ -1,6 +1,6 @@
|
|||||||
FROM bash:alpine3.15
|
FROM bash:alpine3.19
|
||||||
|
|
||||||
RUN apk add parallel aria2
|
RUN apk update && apk add parallel aria2
|
||||||
COPY . /docker
|
COPY . /docker
|
||||||
RUN chmod +x /docker/download.sh
|
RUN chmod +x /docker/download.sh
|
||||||
ENTRYPOINT ["/docker/download.sh"]
|
ENTRYPOINT ["/docker/download.sh"]
|
||||||
|
|||||||
0
services/download/checksums.sha256
Normal file → Executable file
0
services/download/checksums.sha256
Normal file → Executable file
0
services/download/links.txt
Normal file → Executable file
0
services/download/links.txt
Normal file → Executable file
@@ -1,53 +0,0 @@
|
|||||||
FROM alpine:3.17 as xformers
|
|
||||||
RUN apk add --no-cache aria2
|
|
||||||
RUN aria2c -x 5 --dir / --out wheel.whl 'https://github.com/AbdBarho/stable-diffusion-webui-docker/releases/download/6.0.0/xformers-0.0.21.dev544-cp310-cp310-manylinux2014_x86_64-pytorch201.whl'
|
|
||||||
|
|
||||||
|
|
||||||
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
|
|
||||||
|
|
||||||
ENV DEBIAN_FRONTEND=noninteractive PIP_EXISTS_ACTION=w PIP_PREFER_BINARY=1
|
|
||||||
|
|
||||||
# patch match:
|
|
||||||
# https://github.com/invoke-ai/InvokeAI/blob/main/docs/installation/INSTALL_PATCHMATCH.md
|
|
||||||
RUN --mount=type=cache,target=/var/cache/apt \
|
|
||||||
apt-get update && \
|
|
||||||
apt-get install make g++ git libopencv-dev -y && \
|
|
||||||
apt-get clean && \
|
|
||||||
cd /usr/lib/x86_64-linux-gnu/pkgconfig/ && \
|
|
||||||
ln -sf opencv4.pc opencv.pc
|
|
||||||
|
|
||||||
|
|
||||||
ENV ROOT=/InvokeAI
|
|
||||||
RUN git clone https://github.com/invoke-ai/InvokeAI.git ${ROOT}
|
|
||||||
WORKDIR ${ROOT}
|
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
||||||
git reset --hard f3b2e02921927d9317255b1c3811f47bd40a2bf9 && \
|
|
||||||
pip install -e .
|
|
||||||
|
|
||||||
|
|
||||||
ARG BRANCH=main SHA=f3b2e02921927d9317255b1c3811f47bd40a2bf9
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
||||||
git fetch && \
|
|
||||||
git reset --hard && \
|
|
||||||
git checkout ${BRANCH} && \
|
|
||||||
git reset --hard ${SHA} && \
|
|
||||||
pip install -U -e .
|
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
||||||
--mount=type=bind,from=xformers,source=/wheel.whl,target=/xformers-0.0.21-cp310-cp310-linux_x86_64.whl \
|
|
||||||
pip install -U opencv-python-headless triton /xformers-0.0.21-cp310-cp310-linux_x86_64.whl && \
|
|
||||||
python3 -c "from patchmatch import patch_match"
|
|
||||||
|
|
||||||
|
|
||||||
COPY . /docker/
|
|
||||||
|
|
||||||
ENV NVIDIA_VISIBLE_DEVICES=all
|
|
||||||
ENV PYTHONUNBUFFERED=1 PRELOAD=false HF_HOME=/root/.cache/huggingface CONFIG_DIR=/data/config/invoke CLI_ARGS=""
|
|
||||||
EXPOSE 7860
|
|
||||||
|
|
||||||
ENTRYPOINT ["/docker/entrypoint.sh"]
|
|
||||||
CMD invokeai --web --host 0.0.0.0 --port 7860 --root_dir ${ROOT} --config ${CONFIG_DIR}/models.yaml \
|
|
||||||
--outdir /output/invoke --embedding_directory /data/embeddings/ --lora_directory /data/models/Lora \
|
|
||||||
--no-nsfw_checker --no-safety_checker ${CLI_ARGS}
|
|
||||||
|
|
||||||
@@ -1,45 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
|
|
||||||
set -Eeuo pipefail
|
|
||||||
|
|
||||||
declare -A MOUNTS
|
|
||||||
|
|
||||||
mkdir -p ${CONFIG_DIR} ${ROOT}/configs/stable-diffusion/
|
|
||||||
|
|
||||||
# cache
|
|
||||||
MOUNTS["/root/.cache"]=/data/.cache/
|
|
||||||
|
|
||||||
# this is really just a hack to avoid migrations
|
|
||||||
rm -rf ${HF_HOME}/diffusers
|
|
||||||
|
|
||||||
# ui specific
|
|
||||||
MOUNTS["${ROOT}/models/codeformer"]=/data/models/Codeformer/
|
|
||||||
MOUNTS["${ROOT}/models/gfpgan/GFPGANv1.4.pth"]=/data/models/GFPGAN/GFPGANv1.4.pth
|
|
||||||
MOUNTS["${ROOT}/models/gfpgan/weights"]=/data/models/GFPGAN/
|
|
||||||
MOUNTS["${ROOT}/models/realesrgan"]=/data/models/RealESRGAN/
|
|
||||||
|
|
||||||
MOUNTS["${ROOT}/models/ldm"]=/data/.cache/invoke/ldm/
|
|
||||||
|
|
||||||
# hacks
|
|
||||||
|
|
||||||
for to_path in "${!MOUNTS[@]}"; do
|
|
||||||
set -Eeuo pipefail
|
|
||||||
from_path="${MOUNTS[${to_path}]}"
|
|
||||||
rm -rf "${to_path}"
|
|
||||||
mkdir -p "$(dirname "${to_path}")"
|
|
||||||
# ends with slash, make it!
|
|
||||||
if [[ "$from_path" == */ ]]; then
|
|
||||||
mkdir -vp "$from_path"
|
|
||||||
fi
|
|
||||||
|
|
||||||
ln -sT "${from_path}" "${to_path}"
|
|
||||||
echo Mounted $(basename "${from_path}")
|
|
||||||
done
|
|
||||||
|
|
||||||
if "${PRELOAD}" == "true"; then
|
|
||||||
set -Eeuo pipefail
|
|
||||||
invokeai-configure --root ${ROOT} --yes
|
|
||||||
cp ${ROOT}/configs/models.yaml ${CONFIG_DIR}/models.yaml
|
|
||||||
fi
|
|
||||||
|
|
||||||
exec "$@"
|
|
||||||
Reference in New Issue
Block a user