mirror of
https://github.com/AbdBarho/stable-diffusion-webui-docker.git
synced 2025-10-29 17:24:25 -04:00
Compare commits
10 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
110627415d | ||
|
|
87b1509dc2 | ||
|
|
71f4abb2db | ||
|
|
c836f41921 | ||
|
|
d47e77f19c | ||
|
|
7fb8b97b90 | ||
|
|
9ac33db795 | ||
|
|
a68734c9f9 | ||
|
|
36f39043de | ||
|
|
0e5801e9d6 |
1
.github/workflows/docker.yml
vendored
1
.github/workflows/docker.yml
vendored
@@ -14,7 +14,6 @@ jobs:
|
|||||||
matrix:
|
matrix:
|
||||||
profile:
|
profile:
|
||||||
- auto
|
- auto
|
||||||
- sygil
|
|
||||||
- invoke
|
- invoke
|
||||||
- comfy
|
- comfy
|
||||||
- download
|
- download
|
||||||
|
|||||||
4
.gitignore
vendored
4
.gitignore
vendored
@@ -1,2 +1,6 @@
|
|||||||
/.devcontainer
|
/.devcontainer
|
||||||
/docker-compose.override.yml
|
/docker-compose.override.yml
|
||||||
|
|
||||||
|
# VSCode specific
|
||||||
|
*.code-workspace
|
||||||
|
/.vscode
|
||||||
|
|||||||
10
README.md
10
README.md
@@ -26,14 +26,6 @@ This repository provides multiple UIs for you to play around with stable diffusi
|
|||||||
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
||||||
|  |  |  |
|
|  |  |  |
|
||||||
|
|
||||||
### [Sygil (sd-webui / hlky)](https://github.com/Sygil-Dev/sygil-webui)
|
|
||||||
|
|
||||||
[Full feature list here](https://github.com/Sygil-Dev/sygil-webui/blob/master/README.md), Screenshots:
|
|
||||||
|
|
||||||
| Text to image | Image to image | Image Lab |
|
|
||||||
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
|
||||||
|  |  |  |
|
|
||||||
|
|
||||||
### [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,7 +50,7 @@ Special thanks to everyone behind these awesome projects, without them, none of
|
|||||||
|
|
||||||
- [AUTOMATIC1111/stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
|
- [AUTOMATIC1111/stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
|
||||||
- [InvokeAI](https://github.com/invoke-ai/InvokeAI)
|
- [InvokeAI](https://github.com/invoke-ai/InvokeAI)
|
||||||
- [Sygil-webui](https://github.com/Sygil-Dev/sygil-webui)
|
|
||||||
- [ComfyUI](https://github.com/comfyanonymous/ComfyUI)
|
- [ComfyUI](https://github.com/comfyanonymous/ComfyUI)
|
||||||
- [CompVis/stable-diffusion](https://github.com/CompVis/stable-diffusion)
|
- [CompVis/stable-diffusion](https://github.com/CompVis/stable-diffusion)
|
||||||
|
- [Sygil-webui](https://github.com/Sygil-Dev/sygil-webui)
|
||||||
- and many many more.
|
- and many many more.
|
||||||
|
|||||||
2
data/.gitignore
vendored
2
data/.gitignore
vendored
@@ -24,3 +24,5 @@
|
|||||||
/openpose
|
/openpose
|
||||||
/ModelScope
|
/ModelScope
|
||||||
/LyCORIS
|
/LyCORIS
|
||||||
|
/GLIGEN
|
||||||
|
/CLIPEncoder
|
||||||
|
|||||||
@@ -28,7 +28,7 @@ services:
|
|||||||
<<: *base_service
|
<<: *base_service
|
||||||
profiles: ["auto"]
|
profiles: ["auto"]
|
||||||
build: ./services/AUTOMATIC1111
|
build: ./services/AUTOMATIC1111
|
||||||
image: sd-auto:54
|
image: sd-auto:56
|
||||||
environment:
|
environment:
|
||||||
- CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
|
- CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
|
||||||
|
|
||||||
@@ -39,35 +39,27 @@ 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: &invoke
|
||||||
<<: *base_service
|
<<: *base_service
|
||||||
profiles: ["invoke"]
|
profiles: ["invoke"]
|
||||||
build: ./services/invoke/
|
build: ./services/invoke/
|
||||||
image: sd-invoke:27
|
image: sd-invoke:28
|
||||||
environment:
|
environment:
|
||||||
- PRELOAD=true
|
- PRELOAD=true
|
||||||
- CLI_ARGS=--no-nsfw_checker --no-safety_checker --xformers
|
- CLI_ARGS=--xformers
|
||||||
|
|
||||||
sygil: &sygil
|
# invoke-cpu:
|
||||||
<<: *base_service
|
# <<: *invoke
|
||||||
profiles: ["sygil"]
|
# profiles: ["invoke-cpu"]
|
||||||
build: ./services/sygil/
|
# environment:
|
||||||
image: sd-sygil:16
|
# - PRELOAD=true
|
||||||
environment:
|
# - CLI_ARGS=--always_use_cpu
|
||||||
- CLI_ARGS=--optimized-turbo
|
|
||||||
- USE_STREAMLIT=0
|
|
||||||
|
|
||||||
sygil-sl:
|
|
||||||
<<: *sygil
|
|
||||||
profiles: ["sygil-sl"]
|
|
||||||
environment:
|
|
||||||
- USE_STREAMLIT=1
|
|
||||||
|
|
||||||
comfy: &comfy
|
comfy: &comfy
|
||||||
<<: *base_service
|
<<: *base_service
|
||||||
profiles: ["comfy"]
|
profiles: ["comfy"]
|
||||||
build: ./services/comfy/
|
build: ./services/comfy/
|
||||||
image: sd-comfy:1
|
image: sd-comfy:2
|
||||||
tty: true
|
tty: true
|
||||||
environment:
|
environment:
|
||||||
- CLI_ARGS=
|
- CLI_ARGS=
|
||||||
|
|||||||
@@ -18,17 +18,20 @@ RUN . /clone.sh clip-interrogator https://github.com/pharmapsychotic/clip-interr
|
|||||||
|
|
||||||
FROM alpine:3.17 as xformers
|
FROM alpine:3.17 as xformers
|
||||||
RUN apk add --no-cache aria2
|
RUN apk add --no-cache aria2
|
||||||
RUN aria2c -x 5 --dir / --out wheel.whl 'https://github.com/AbdBarho/stable-diffusion-webui-docker/releases/download/5.0.0/xformers-0.0.17.dev449-cp310-cp310-manylinux2014_x86_64.whl'
|
RUN aria2c -x 5 --dir / --out wheel.whl 'https://github.com/AbdBarho/stable-diffusion-webui-docker/releases/download/5.0.3/xformers-0.0.20.dev528-cp310-cp310-manylinux2014_x86_64-pytorch2.whl'
|
||||||
|
|
||||||
|
|
||||||
FROM python:3.10.9-slim
|
FROM python:3.10.9-slim
|
||||||
|
|
||||||
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
|
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip pip install torch==1.13.1+cu117 torchvision --extra-index-url https://download.pytorch.org/whl/cu117
|
RUN apt-get update && apt install fonts-dejavu-core rsync git jq moreutils aria2 -y && apt-get clean
|
||||||
|
|
||||||
# RUN --mount=type=cache,target=/root/.cache/pip pip install torch==2.0.0+cu118 torchvision --extra-index-url https://download.pytorch.org/whl/cu117
|
RUN --mount=type=cache,target=/cache --mount=type=cache,target=/root/.cache/pip \
|
||||||
|
aria2c -x 5 --dir /cache --out torch-2.0.0-cp310-cp310-linux_x86_64.whl -c \
|
||||||
|
https://download.pytorch.org/whl/cu118/torch-2.0.0%2Bcu118-cp310-cp310-linux_x86_64.whl && \
|
||||||
|
pip install /cache/torch-2.0.0-cp310-cp310-linux_x86_64.whl torchvision --index-url https://download.pytorch.org/whl/cu118
|
||||||
|
|
||||||
RUN apt-get update && apt install fonts-dejavu-core rsync git jq moreutils -y && apt-get clean
|
|
||||||
|
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
||||||
@@ -38,8 +41,8 @@ RUN --mount=type=cache,target=/root/.cache/pip \
|
|||||||
pip install -r requirements_versions.txt
|
pip install -r requirements_versions.txt
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
||||||
--mount=type=bind,from=xformers,source=/wheel.whl,target=/xformers-0.0.15-cp310-cp310-linux_x86_64.whl \
|
--mount=type=bind,from=xformers,source=/wheel.whl,target=/xformers-0.0.20.dev528-cp310-cp310-manylinux2014_x86_64.whl \
|
||||||
pip install triton /xformers-0.0.15-cp310-cp310-linux_x86_64.whl
|
pip install /xformers-0.0.20.dev528-cp310-cp310-manylinux2014_x86_64.whl
|
||||||
|
|
||||||
ENV ROOT=/stable-diffusion-webui
|
ENV ROOT=/stable-diffusion-webui
|
||||||
|
|
||||||
@@ -62,7 +65,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
|
|||||||
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
|
||||||
|
|
||||||
ARG SHA=5ab7f213bec2f816f9c5644becb32eb72c8ffb89
|
ARG SHA=89f9faa63388756314e8a1d96cf86bf5e0663045
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
||||||
cd stable-diffusion-webui && \
|
cd stable-diffusion-webui && \
|
||||||
git fetch && \
|
git fetch && \
|
||||||
|
|||||||
@@ -1,10 +0,0 @@
|
|||||||
{
|
|
||||||
"outdir_samples": "",
|
|
||||||
"outdir_txt2img_samples": "/output/txt2img",
|
|
||||||
"outdir_img2img_samples": "/output/img2img",
|
|
||||||
"outdir_extras_samples": "/output/extras",
|
|
||||||
"outdir_txt2img_grids": "/output/txt2img-grids",
|
|
||||||
"outdir_img2img_grids": "/output/img2img-grids",
|
|
||||||
"outdir_save": "/output/saved",
|
|
||||||
"font": "DejaVuSans.ttf"
|
|
||||||
}
|
|
||||||
78
services/AUTOMATIC1111/config.py
Normal file
78
services/AUTOMATIC1111/config.py
Normal 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/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)
|
||||||
|
|
||||||
@@ -8,8 +8,8 @@ mkdir -p /data/config/auto/scripts/
|
|||||||
find "${ROOT}/scripts/" -maxdepth 1 -type l -delete
|
find "${ROOT}/scripts/" -maxdepth 1 -type l -delete
|
||||||
cp -vrfTs /data/config/auto/scripts/ "${ROOT}/scripts/"
|
cp -vrfTs /data/config/auto/scripts/ "${ROOT}/scripts/"
|
||||||
|
|
||||||
cp -n /docker/config.json /data/config/auto/config.json
|
# Set up config file
|
||||||
jq '. * input' /data/config/auto/config.json /docker/config.json | sponge /data/config/auto/config.json
|
python /docker/config.py /data/config/auto/config.json
|
||||||
|
|
||||||
if [ ! -f /data/config/auto/ui-config.json ]; then
|
if [ ! -f /data/config/auto/ui-config.json ]; then
|
||||||
echo '{}' >/data/config/auto/ui-config.json
|
echo '{}' >/data/config/auto/ui-config.json
|
||||||
|
|||||||
@@ -36,6 +36,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
|
|||||||
|
|
||||||
# add info
|
# add info
|
||||||
COPY . /docker/
|
COPY . /docker/
|
||||||
|
RUN cp /docker/extra_model_paths.yaml ${ROOT}
|
||||||
|
|
||||||
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility NVIDIA_VISIBLE_DEVICES=all
|
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility NVIDIA_VISIBLE_DEVICES=all
|
||||||
ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
|
ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
|
||||||
|
|||||||
@@ -2,34 +2,12 @@
|
|||||||
|
|
||||||
set -Eeuo pipefail
|
set -Eeuo pipefail
|
||||||
|
|
||||||
|
mkdir -vp /data/config/comfy/custom_nodes
|
||||||
|
|
||||||
declare -A MOUNTS
|
declare -A MOUNTS
|
||||||
|
|
||||||
|
MOUNTS["/root/.cache"]="/data/.cache"
|
||||||
mkdir -vp /data/config/comfy/
|
MOUNTS["${ROOT}/input"]="/data/config/comfy/input"
|
||||||
|
|
||||||
# cache
|
|
||||||
MOUNTS["/root/.cache"]=/data/.cache
|
|
||||||
# ui specific
|
|
||||||
MOUNTS["${ROOT}/models/checkpoints"]="/data/StableDiffusion"
|
|
||||||
MOUNTS["${ROOT}/models/controlnet"]="/data/ControlNet"
|
|
||||||
MOUNTS["${ROOT}/models/upscale_models/RealESRGAN"]="/data/RealESRGAN"
|
|
||||||
MOUNTS["${ROOT}/models/upscale_models/GFPGAN"]="/data/GFPGAN"
|
|
||||||
MOUNTS["${ROOT}/models/upscale_models/SwinIR"]="/data/SwinIR"
|
|
||||||
MOUNTS["${ROOT}/models/vae"]="/data/VAE"
|
|
||||||
|
|
||||||
# data
|
|
||||||
MOUNTS["${ROOT}/models/loras"]="/data/Lora"
|
|
||||||
MOUNTS["${ROOT}/models/embeddings"]="/data/embeddings"
|
|
||||||
|
|
||||||
# config
|
|
||||||
# TODO: I am not sure if this is final, maybe it should change in the future
|
|
||||||
MOUNTS["${ROOT}/models/clip"]="/data/.cache/comfy/clip"
|
|
||||||
MOUNTS["${ROOT}/models/clip_vision"]="/data/.cache/comfy/clip_vision"
|
|
||||||
MOUNTS["${ROOT}/models/custom_nodes"]="/data/config/comfy/custom_nodes"
|
|
||||||
MOUNTS["${ROOT}/models/style_models"]="/data/config/comfy/style_models"
|
|
||||||
MOUNTS["${ROOT}/models/t2i_adapter"]="/data/config/comfy/t2i_adapter"
|
|
||||||
|
|
||||||
# output
|
|
||||||
MOUNTS["${ROOT}/output"]="/output/comfy"
|
MOUNTS["${ROOT}/output"]="/output/comfy"
|
||||||
|
|
||||||
for to_path in "${!MOUNTS[@]}"; do
|
for to_path in "${!MOUNTS[@]}"; do
|
||||||
|
|||||||
25
services/comfy/extra_model_paths.yaml
Normal file
25
services/comfy/extra_model_paths.yaml
Normal file
@@ -0,0 +1,25 @@
|
|||||||
|
a111:
|
||||||
|
base_path: /data
|
||||||
|
|
||||||
|
checkpoints: StableDiffusion
|
||||||
|
configs: StableDiffusion
|
||||||
|
vae: VAE
|
||||||
|
loras: Lora
|
||||||
|
upscale_models: |
|
||||||
|
RealESRGAN
|
||||||
|
ESRGAN
|
||||||
|
SwinIR
|
||||||
|
GFPGAN
|
||||||
|
embeddings: embeddings
|
||||||
|
hypernetworks: Hypernetworks
|
||||||
|
controlnet: ControlNet
|
||||||
|
gligen: GLIGEN
|
||||||
|
clip: CLIPEncoder
|
||||||
|
|
||||||
|
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
|
||||||
@@ -3,7 +3,27 @@
|
|||||||
set -Eeuo pipefail
|
set -Eeuo pipefail
|
||||||
|
|
||||||
# TODO: maybe just use the .gitignore file to create all of these
|
# TODO: maybe just use the .gitignore file to create all of these
|
||||||
mkdir -vp /data/.cache /data/StableDiffusion /data/LyCORIS /data/Codeformer /data/ModelScope /data/GFPGAN /data/ESRGAN /data/BSRGAN /data/RealESRGAN /data/SwinIR /data/LDSR /data/ScuNET /data/embeddings /data/VAE /data/Deepdanbooru /data/MiDaS /data/Lora /data/ControlNet /data/openpose
|
mkdir -vp /data/.cache \
|
||||||
|
/data/StableDiffusion \
|
||||||
|
/data/LyCORIS \
|
||||||
|
/data/Codeformer \
|
||||||
|
/data/ModelScope \
|
||||||
|
/data/GFPGAN \
|
||||||
|
/data/ESRGAN \
|
||||||
|
/data/BSRGAN \
|
||||||
|
/data/RealESRGAN \
|
||||||
|
/data/SwinIR \
|
||||||
|
/data/LDSR \
|
||||||
|
/data/ScuNET \
|
||||||
|
/data/embeddings \
|
||||||
|
/data/VAE \
|
||||||
|
/data/Deepdanbooru \
|
||||||
|
/data/MiDaS \
|
||||||
|
/data/Lora \
|
||||||
|
/data/ControlNet \
|
||||||
|
/data/openpose \
|
||||||
|
/data/GLIGEN \
|
||||||
|
/data/CLIPEncoder # this is different from the cached pre-train ViT saved in /data/.cache/clip TODO: find out how
|
||||||
|
|
||||||
echo "Downloading, this might take a while..."
|
echo "Downloading, this might take a while..."
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
FROM alpine:3.17 as xformers
|
FROM alpine:3.17 as xformers
|
||||||
RUN apk add --no-cache aria2
|
RUN apk add --no-cache aria2
|
||||||
RUN aria2c -x 5 --dir / --out wheel.whl 'https://github.com/AbdBarho/stable-diffusion-webui-docker/releases/download/5.0.0/xformers-0.0.17.dev449-cp310-cp310-manylinux2014_x86_64.whl'
|
RUN aria2c -x 5 --dir / --out wheel.whl 'https://github.com/AbdBarho/stable-diffusion-webui-docker/releases/download/5.0.3/xformers-0.0.20.dev528-cp310-cp310-manylinux2014_x86_64-pytorch1.13.whl'
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@@ -22,27 +22,26 @@ RUN --mount=type=cache,target=/var/cache/apt \
|
|||||||
ln -sf opencv4.pc opencv.pc
|
ln -sf opencv4.pc opencv.pc
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
ENV ROOT=/InvokeAI
|
ENV ROOT=/InvokeAI
|
||||||
RUN git clone https://github.com/invoke-ai/InvokeAI.git ${ROOT}
|
RUN git clone https://github.com/invoke-ai/InvokeAI.git ${ROOT}
|
||||||
WORKDIR ${ROOT}
|
WORKDIR ${ROOT}
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
||||||
git reset --hard 4463124bddd221c333d4c70e73aa2949ad35453d && \
|
git reset --hard 4463124bddd221c333d4c70e73aa2949ad35453d && \
|
||||||
pip install .
|
pip install -e .
|
||||||
|
|
||||||
|
|
||||||
ARG BRANCH=main SHA=50eb02f68be912276a9c106d5e8038a5671a0386
|
ARG BRANCH=main SHA=d73f1c363c3f2fe00f14ed34e276b4366b32a886
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
||||||
git fetch && \
|
git fetch && \
|
||||||
git reset --hard && \
|
git reset --hard && \
|
||||||
git checkout ${BRANCH} && \
|
git checkout ${BRANCH} && \
|
||||||
git reset --hard ${SHA} && \
|
git reset --hard ${SHA} && \
|
||||||
pip install -U .
|
pip install -U -e .
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
||||||
--mount=type=bind,from=xformers,source=/wheel.whl,target=/xformers-0.0.17-cp310-cp310-linux_x86_64.whl \
|
--mount=type=bind,from=xformers,source=/wheel.whl,target=/xformers-0.0.20-cp310-cp310-linux_x86_64.whl \
|
||||||
pip install -U opencv-python-headless triton /xformers-0.0.17-cp310-cp310-linux_x86_64.whl && \
|
pip install -U opencv-python-headless triton /xformers-0.0.20-cp310-cp310-linux_x86_64.whl && \
|
||||||
python3 -c "from patchmatch import patch_match"
|
python3 -c "from patchmatch import patch_match"
|
||||||
|
|
||||||
|
|
||||||
@@ -54,5 +53,7 @@ ENV PYTHONUNBUFFERED=1 PRELOAD=false HF_HOME=/root/.cache/huggingface CONFIG_DIR
|
|||||||
EXPOSE 7860
|
EXPOSE 7860
|
||||||
|
|
||||||
ENTRYPOINT ["/docker/entrypoint.sh"]
|
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 ${CLI_ARGS}
|
CMD invokeai --web --host 0.0.0.0 --port 7860 --root_dir ${ROOT} --config ${CONFIG_DIR}/models.yaml \
|
||||||
# TODO: make sure the config is persisted between sessions
|
--outdir /output/invoke --embedding_directory /data/embeddings/ --lora_directory /data/Lora \
|
||||||
|
--no-nsfw_checker --no-safety_checker ${CLI_ARGS}
|
||||||
|
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ set -Eeuo pipefail
|
|||||||
|
|
||||||
declare -A MOUNTS
|
declare -A MOUNTS
|
||||||
|
|
||||||
mkdir -p ${CONFIG_DIR}
|
mkdir -p ${CONFIG_DIR} ${ROOT}/configs/stable-diffusion/
|
||||||
|
|
||||||
# cache
|
# cache
|
||||||
MOUNTS["/root/.cache"]=/data/.cache/
|
MOUNTS["/root/.cache"]=/data/.cache/
|
||||||
@@ -20,8 +20,6 @@ MOUNTS["${ROOT}/models/realesrgan"]=/data/RealESRGAN/
|
|||||||
|
|
||||||
MOUNTS["${ROOT}/models/ldm"]=/data/.cache/invoke/ldm/
|
MOUNTS["${ROOT}/models/ldm"]=/data/.cache/invoke/ldm/
|
||||||
|
|
||||||
MOUNTS["${ROOT}/embeddings"]=/data/embeddings/
|
|
||||||
|
|
||||||
# hacks
|
# hacks
|
||||||
|
|
||||||
for to_path in "${!MOUNTS[@]}"; do
|
for to_path in "${!MOUNTS[@]}"; do
|
||||||
|
|||||||
@@ -1,40 +0,0 @@
|
|||||||
FROM python:3.8-slim
|
|
||||||
|
|
||||||
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
|
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip pip install torch==1.13.0 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
|
|
||||||
|
|
||||||
RUN apt-get update && apt install gcc libsndfile1 ffmpeg build-essential zip unzip git -y && apt-get clean
|
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
||||||
git config --global http.postBuffer 1048576000 && \
|
|
||||||
git clone https://github.com/Sygil-Dev/sygil-webui.git stable-diffusion && \
|
|
||||||
cd stable-diffusion && \
|
|
||||||
git reset --hard 5291437085bddd16d752f811b6552419a2044d12 && \
|
|
||||||
pip install -r requirements.txt
|
|
||||||
|
|
||||||
|
|
||||||
ARG BRANCH=master SHA=571fb897edd58b714bb385dfaa1ad59aecef8bc7
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
||||||
cd stable-diffusion && \
|
|
||||||
git fetch && \
|
|
||||||
git checkout ${BRANCH} && \
|
|
||||||
git reset --hard ${SHA} && \
|
|
||||||
pip install -r requirements.txt
|
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip pip install -U 'transformers>=4.24'
|
|
||||||
|
|
||||||
# add info
|
|
||||||
COPY . /docker/
|
|
||||||
RUN python /docker/info.py /stable-diffusion/frontend/frontend.py && \
|
|
||||||
chmod +x /docker/mount.sh /docker/run.sh && \
|
|
||||||
# streamlit \
|
|
||||||
sed -i -- 's/8501/7860/g' /stable-diffusion/.streamlit/config.toml
|
|
||||||
|
|
||||||
WORKDIR /stable-diffusion
|
|
||||||
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
|
||||||
ENV NVIDIA_VISIBLE_DEVICES=all
|
|
||||||
ENV PYTHONPATH="${PYTHONPATH}:${PWD}" STREAMLIT_SERVER_HEADLESS=true USE_STREAMLIT=0 CLI_ARGS=""
|
|
||||||
EXPOSE 7860
|
|
||||||
|
|
||||||
CMD /docker/mount.sh && /docker/run.sh
|
|
||||||
@@ -1,13 +0,0 @@
|
|||||||
import sys
|
|
||||||
from pathlib import Path
|
|
||||||
|
|
||||||
file = Path(sys.argv[1])
|
|
||||||
file.write_text(
|
|
||||||
file.read_text()\
|
|
||||||
.replace('<p>For help and advanced usage guides,', """
|
|
||||||
<p>
|
|
||||||
Created using <a href="https://github.com/AbdBarho/stable-diffusion-webui-docker">stable-diffusion-webui-docker</a>.
|
|
||||||
</p>
|
|
||||||
<p>For help and advanced usage guides,
|
|
||||||
""", 1)
|
|
||||||
)
|
|
||||||
@@ -1,32 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
|
|
||||||
set -Eeuo pipefail
|
|
||||||
|
|
||||||
declare -A MOUNTS
|
|
||||||
|
|
||||||
ROOT=/stable-diffusion/src
|
|
||||||
|
|
||||||
# cache
|
|
||||||
MOUNTS["/root/.cache"]=/data/.cache
|
|
||||||
# ui specific
|
|
||||||
MOUNTS["${PWD}/models/realesrgan"]=/data/RealESRGAN
|
|
||||||
MOUNTS["${PWD}/models/ldsr"]=/data/LDSR
|
|
||||||
MOUNTS["${PWD}/models/custom"]=/data/StableDiffusion
|
|
||||||
|
|
||||||
# hack
|
|
||||||
MOUNTS["${PWD}/models/gfpgan/GFPGANv1.3.pth"]=/data/GFPGAN/GFPGANv1.4.pth
|
|
||||||
MOUNTS["${PWD}/models/gfpgan/GFPGANv1.4.pth"]=/data/GFPGAN/GFPGANv1.4.pth
|
|
||||||
MOUNTS["${PWD}/gfpgan/weights"]=/data/.cache
|
|
||||||
|
|
||||||
|
|
||||||
for to_path in "${!MOUNTS[@]}"; do
|
|
||||||
set -Eeuo pipefail
|
|
||||||
from_path="${MOUNTS[${to_path}]}"
|
|
||||||
rm -rf "${to_path}"
|
|
||||||
mkdir -p "$(dirname "${to_path}")"
|
|
||||||
ln -sT "${from_path}" "${to_path}"
|
|
||||||
echo Mounted $(basename "${from_path}")
|
|
||||||
done
|
|
||||||
|
|
||||||
# streamlit config
|
|
||||||
ln -sf /docker/userconfig_streamlit.yaml /stable-diffusion/configs/webui/userconfig_streamlit.yaml
|
|
||||||
@@ -1,10 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
|
|
||||||
set -Eeuo pipefail
|
|
||||||
|
|
||||||
echo "USE_STREAMLIT = ${USE_STREAMLIT}"
|
|
||||||
if [ "${USE_STREAMLIT}" == "1" ]; then
|
|
||||||
python -u -m streamlit run scripts/webui_streamlit.py
|
|
||||||
else
|
|
||||||
python3 -u scripts/webui.py --outdir /output --ckpt /data/StableDiffusion/v1-5-pruned-emaonly.ckpt ${CLI_ARGS}
|
|
||||||
fi
|
|
||||||
@@ -1,11 +0,0 @@
|
|||||||
# https://github.com/Sygil-Dev/sygil-webui/blob/master/configs/webui/webui_streamlit.yaml
|
|
||||||
general:
|
|
||||||
version: 1.24.6
|
|
||||||
outdir: /output
|
|
||||||
default_model: "Stable Diffusion v1.5"
|
|
||||||
default_model_path: /data/StableDiffusion/v1-5-pruned-emaonly.ckpt
|
|
||||||
outdir_txt2img: /output/txt2img
|
|
||||||
outdir_img2img: /output/img2img
|
|
||||||
outdir_img2txt: /output/img2txt
|
|
||||||
optimized: True
|
|
||||||
optimized_turbo: True
|
|
||||||
Reference in New Issue
Block a user