mirror of
https://github.com/AbdBarho/stable-diffusion-webui-docker.git
synced 2025-10-27 08:14:26 -04:00
Compare commits
3 Commits
246402e84d
...
9.0.0
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
802d0bcd68 | ||
|
|
b1a26b8041 | ||
|
|
f1bf3b0943 |
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:
|
||||||
|
|||||||
@@ -27,7 +27,7 @@ services:
|
|||||||
<<: *base_service
|
<<: *base_service
|
||||||
profiles: ["auto"]
|
profiles: ["auto"]
|
||||||
build: ./services/AUTOMATIC1111
|
build: ./services/AUTOMATIC1111
|
||||||
image: sd-auto:76
|
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
|
||||||
|
|
||||||
@@ -38,22 +38,6 @@ 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
|
|
||||||
<<: *base_service
|
|
||||||
profiles: ["invoke"]
|
|
||||||
build: ./services/invoke/
|
|
||||||
image: sd-invoke:30
|
|
||||||
environment:
|
|
||||||
- PRELOAD=true
|
|
||||||
- CLI_ARGS=--xformers
|
|
||||||
|
|
||||||
# invoke-cpu:
|
|
||||||
# <<: *invoke
|
|
||||||
# profiles: ["invoke-cpu"]
|
|
||||||
# environment:
|
|
||||||
# - PRELOAD=true
|
|
||||||
# - CLI_ARGS=--always_use_cpu
|
|
||||||
|
|
||||||
comfy: &comfy
|
comfy: &comfy
|
||||||
<<: *base_service
|
<<: *base_service
|
||||||
profiles: ["comfy"]
|
profiles: ["comfy"]
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ RUN . /clone.sh generative-models https://github.com/Stability-AI/generative-mod
|
|||||||
RUN . /clone.sh stable-diffusion-webui-assets https://github.com/AUTOMATIC1111/stable-diffusion-webui-assets 6f7db241d2f8ba7457bac5ca9753331f0c266917
|
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
|
||||||
|
|
||||||
@@ -30,7 +30,7 @@ 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 v1.9.3 && \
|
git reset --hard v1.9.4 && \
|
||||||
pip install -r requirements_versions.txt
|
pip install -r requirements_versions.txt
|
||||||
|
|
||||||
|
|
||||||
@@ -40,7 +40,7 @@ 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 \
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
||||||
pip install pyngrok xformers==0.0.23.post1 \
|
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
|
||||||
|
|||||||
@@ -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