mirror of
https://github.com/AbdBarho/stable-diffusion-webui-docker.git
synced 2025-10-27 16:24:26 -04:00
Compare commits
14 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
3c544dd7f4 | ||
|
|
42cc17da74 | ||
|
|
31e4dec08f | ||
|
|
0148e5e109 | ||
|
|
111825ac25 | ||
|
|
c1e13867d9 | ||
|
|
463f332d14 | ||
|
|
3682303355 | ||
|
|
402c691a49 | ||
|
|
b36113b7d8 | ||
|
|
b60c787474 | ||
|
|
161fd52c16 | ||
|
|
3b3c244c31 | ||
|
|
5698c49653 |
27
.github/ISSUE_TEMPLATE/bug.md
vendored
27
.github/ISSUE_TEMPLATE/bug.md
vendored
@@ -7,30 +7,39 @@ assignees: ''
|
||||
|
||||
---
|
||||
|
||||
**Has this issue been opened before? Check the [FAQ](https://github.com/AbdBarho/stable-diffusion-webui-docker/wiki/Main), the [issues](https://github.com/AbdBarho/stable-diffusion-webui-docker/issues?q=is%3Aissue)**
|
||||
<!-- 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**
|
||||
|
||||
hlky or auto or auto-cpu or lstein?
|
||||
|
||||
|
||||
**Hardware / Software**
|
||||
- OS: [e.g. Windows 10 / Ubuntu ]
|
||||
- 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
|
||||
|
||||
**Hardware / Software:**
|
||||
- OS: [e.g. Windows / Ubuntu and version]
|
||||
- RAM:
|
||||
- GPU: [Nvidia 1660 / No GPU]
|
||||
- VRAM:
|
||||
- Docker Version, Docker compose version
|
||||
- Release version [e.g. 1.0.1]
|
||||
|
||||
**Additional context**
|
||||
Any other context about the problem here. If applicable, add screenshots to help explain your problem.
|
||||
|
||||
2
.github/pull_request_template.md
vendored
2
.github/pull_request_template.md
vendored
@@ -2,4 +2,4 @@
|
||||
|
||||
- auto: https://github.com/AUTOMATIC1111/stable-diffusion-webui/commit/
|
||||
- hlky: https://github.com/sd-webui/stable-diffusion-webui/commit/
|
||||
- lstein: https://github.com/lstein/stable-diffusion/commit/
|
||||
- lstein: https://github.com/invoke-ai/InvokeAI/commit/
|
||||
|
||||
22
.github/workflows/executable.yml1
vendored
22
.github/workflows/executable.yml1
vendored
@@ -1,22 +0,0 @@
|
||||
name: Check executable
|
||||
|
||||
on: [push]
|
||||
|
||||
jobs:
|
||||
check:
|
||||
runs-on: ubuntu-latest
|
||||
name: Check all sh
|
||||
steps:
|
||||
- run: git config --global core.fileMode true
|
||||
- uses: actions/checkout@v3
|
||||
- shell: bash
|
||||
run: |
|
||||
shopt -s globstar;
|
||||
FAIL=0
|
||||
for file in **/*.sh; do
|
||||
if [ -f "${file}" ] && [ -r "${file}" ] && [ ! -x "${file}" ]; then
|
||||
echo "$file" is not executable;
|
||||
FAIL=1
|
||||
fi
|
||||
done
|
||||
exit ${FAIL}
|
||||
36
.github/workflows/xformers.yml
vendored
Normal file
36
.github/workflows/xformers.yml
vendored
Normal file
@@ -0,0 +1,36 @@
|
||||
name: Build Xformers
|
||||
|
||||
on:
|
||||
workflow_dispatch: {}
|
||||
|
||||
jobs:
|
||||
build:
|
||||
runs-on: ubuntu-latest
|
||||
timeout-minutes: 180
|
||||
container:
|
||||
image: python:3.10-slim
|
||||
env:
|
||||
DEBIAN_FRONTEND: noninteractive
|
||||
XFORMERS_DISABLE_FLASH_ATTN: 1
|
||||
FORCE_CUDA: 1
|
||||
TORCH_CUDA_ARCH_LIST: "6.0;6.1;6.2;7.0;7.2;7.5;8.0;8.6"
|
||||
NVCC_FLAGS: --use_fast_math -DXFORMERS_MEM_EFF_ATTENTION_DISABLE_BACKWARD
|
||||
MAX_JOBS: 4
|
||||
steps:
|
||||
- run: |
|
||||
apt-get update
|
||||
apt-get install gpg wget git -y
|
||||
wget https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.0-1_all.deb
|
||||
dpkg -i cuda-keyring_1.0-1_all.deb
|
||||
apt-get update
|
||||
apt-get install cuda-nvcc-11-8 cuda-libraries-dev-11-8 -y
|
||||
|
||||
export PIP_CACHE_DIR=$(pwd)/cache
|
||||
pip install ninja install torch --extra-index-url https://download.pytorch.org/whl/cu113
|
||||
|
||||
pip wheel --wheel-dir=data git+https://github.com/facebookresearch/xformers.git@3633e1afc7bffbe61957f04e7bb1a742ee910ace#egg=xformers
|
||||
- name: Artifacts
|
||||
uses: actions/upload-artifact@v3
|
||||
with:
|
||||
name: xformers
|
||||
path: data/xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl
|
||||
6
LICENSE
6
LICENSE
@@ -92,5 +92,11 @@ use).
|
||||
|
||||
By using this software, you also agree to the following licenses:
|
||||
https://github.com/CompVis/stable-diffusion/blob/main/LICENSE
|
||||
https://github.com/sd-webui/stable-diffusion-webui/blob/master/LICENSE
|
||||
https://github.com/invoke-ai/InvokeAI/blob/main/LICENSE
|
||||
https://github.com/cszn/BSRGAN/blob/main/LICENSE
|
||||
https://github.com/sczhou/CodeFormer/blob/master/LICENSE
|
||||
https://github.com/TencentARC/GFPGAN/blob/master/LICENSE
|
||||
https://github.com/xinntao/Real-ESRGAN/blob/master/LICENSE
|
||||
https://github.com/xinntao/ESRGAN/blob/master/LICENSE
|
||||
https://github.com/cszn/SCUNet/blob/main/LICENSE
|
||||
|
||||
19
README.md
19
README.md
@@ -23,9 +23,9 @@ This repository provides multiple UIs for you to play around with stable diffusi
|
||||
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
||||
|  |  |  |
|
||||
|
||||
### hlky
|
||||
### hlky (sd-webui)
|
||||
|
||||
[hlky's fork](https://github.com/hlky/stable-diffusion-webui) is one of the most popular UIs, with many features:
|
||||
[hlky's fork](https://github.com/sd-webui/stable-diffusion-webui) is one of the most popular UIs, with many features:
|
||||
|
||||
- Text to image, with many samplers
|
||||
- Image to image, with masking, cropping, in-painting, variations.
|
||||
@@ -40,15 +40,20 @@ Screenshots:
|
||||
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
||||
|  |  |  |
|
||||
|
||||
<!--
|
||||
### lstein
|
||||
|
||||
[lstein's fork](https://github.com/lstein/stable-diffusion) is very mature when it comes to the cli, and the WebUI has potential.
|
||||
|
||||
### lstein (InvokeAI)
|
||||
|
||||
[lstein's fork](https://github.com/invoke-ai/InvokeAI) is one of the earliest with a wonderful WebUI.
|
||||
- Text to image, with many samplers
|
||||
- Image to image
|
||||
- 4GB GPU support
|
||||
- More coming!
|
||||
- [Full feature list here](https://github.com/invoke-ai/InvokeAI#features)
|
||||
|
||||
| Text to image | Image to image | Extras |
|
||||
| ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------- |
|
||||
|  |  |  |
|
||||
-->
|
||||
|  |  |  |
|
||||
|
||||
## Setup & Usage
|
||||
|
||||
|
||||
3
data/.gitignore
vendored
3
data/.gitignore
vendored
@@ -1,5 +1,7 @@
|
||||
# for all of the stuff downloaded by transformers, pytorch, and others
|
||||
/.cache
|
||||
# for UIs
|
||||
/config
|
||||
# for all stable diffusion models (main, waifu diffusion, etc..)
|
||||
/StableDiffusion
|
||||
# others
|
||||
@@ -11,4 +13,5 @@
|
||||
/SwinIR
|
||||
/ScuNET
|
||||
/LDSR
|
||||
/Hypernetworks
|
||||
/embeddings
|
||||
|
||||
@@ -27,19 +27,18 @@ services:
|
||||
<<: *base_service
|
||||
profiles: ["hlky"]
|
||||
build: ./services/hlky/
|
||||
image: sd-hlky:1
|
||||
environment:
|
||||
- CLI_ARGS=--optimized-turbo
|
||||
- USE_STREAMLIT=0
|
||||
|
||||
auto: &automatic
|
||||
<<: *base_service
|
||||
profiles: ["auto"]
|
||||
build: ./services/AUTOMATIC1111
|
||||
volumes:
|
||||
- *v1
|
||||
- *v2
|
||||
- ./services/AUTOMATIC1111/config.json:/stable-diffusion-webui/config.json
|
||||
image: sd-auto:1
|
||||
environment:
|
||||
- CLI_ARGS=--allow-code --medvram
|
||||
- CLI_ARGS=--allow-code --medvram --xformers
|
||||
|
||||
auto-cpu:
|
||||
<<: *automatic
|
||||
@@ -52,6 +51,7 @@ services:
|
||||
<<: *base_service
|
||||
profiles: ["lstein"]
|
||||
build: ./services/lstein/
|
||||
image: sd-lstein:1
|
||||
environment:
|
||||
- PRELOAD=false
|
||||
- PRELOAD=true
|
||||
- CLI_ARGS=
|
||||
|
||||
@@ -2,41 +2,45 @@
|
||||
|
||||
FROM alpine/git:2.36.2 as download
|
||||
|
||||
SHELL ["/bin/sh", "-ceuxo", "pipefail"]
|
||||
|
||||
RUN git clone https://github.com/CompVis/stable-diffusion.git repositories/stable-diffusion && cd repositories/stable-diffusion && git reset --hard 69ae4b35e0a0f6ee1af8bb9a5d0016ccb27e36dc
|
||||
|
||||
RUN git clone https://github.com/sczhou/CodeFormer.git repositories/CodeFormer && cd repositories/CodeFormer && git reset --hard c5b4593074ba6214284d6acd5f1719b6c5d739af
|
||||
RUN git clone https://github.com/salesforce/BLIP.git repositories/BLIP && cd repositories/BLIP && git reset --hard 48211a1594f1321b00f14c9f7a5b4813144b2fb9
|
||||
RUN git clone https://github.com/Hafiidz/latent-diffusion.git repositories/latent-diffusion && cd repositories/latent-diffusion && git reset --hard abf33e7002d59d9085081bce93ec798dcabd49af
|
||||
|
||||
RUN <<EOF
|
||||
# because taming-transformers is huge
|
||||
git config --global http.postBuffer 1048576000
|
||||
git clone https://github.com/CompVis/taming-transformers.git repositories/taming-transformers
|
||||
cd repositories/taming-transformers
|
||||
git reset --hard 24268930bf1dce879235a7fddd0b2355b84d7ea6
|
||||
rm -rf repositories/taming-transformers/data repositories/taming-transformers/assets
|
||||
rm -rf data assets
|
||||
EOF
|
||||
|
||||
RUN git clone https://github.com/crowsonkb/k-diffusion.git repositories/k-diffusion && cd repositories/k-diffusion && git reset --hard f4e99857772fc3a126ba886aadf795a332774878
|
||||
|
||||
|
||||
FROM continuumio/miniconda3:4.12.0
|
||||
FROM alpine:3 as xformers
|
||||
RUN apk add aria2
|
||||
RUN aria2c --dir / --out wheel.whl 'https://github.com/AbdBarho/stable-diffusion-webui-docker/releases/download/2.1.0/xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl'
|
||||
|
||||
FROM python:3.10-slim
|
||||
|
||||
SHELL ["/bin/bash", "-ceuxo", "pipefail"]
|
||||
|
||||
ENV DEBIAN_FRONTEND=noninteractive
|
||||
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1 PIP_NO_CACHE_DIR=1
|
||||
|
||||
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 pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113
|
||||
|
||||
RUN apt-get update && apt install fonts-dejavu-core rsync -y && apt-get clean
|
||||
RUN apt-get update && apt install fonts-dejavu-core rsync git -y && apt-get clean
|
||||
|
||||
|
||||
RUN <<EOF
|
||||
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
|
||||
cd stable-diffusion-webui
|
||||
git reset --hard 1eb588cbf19924333b88beaa1ac0041904966640
|
||||
pip install --prefer-binary --no-cache-dir -r requirements_versions.txt
|
||||
pip install -r requirements_versions.txt
|
||||
EOF
|
||||
|
||||
ENV ROOT=/stable-diffusion-webui \
|
||||
@@ -46,32 +50,38 @@ ENV ROOT=/stable-diffusion-webui \
|
||||
COPY --from=download /git/ ${ROOT}
|
||||
RUN pip install --prefer-binary --no-cache-dir -r ${ROOT}/repositories/CodeFormer/requirements.txt
|
||||
|
||||
# TODO: move to top
|
||||
RUN apt-get install jq moreutils -y
|
||||
|
||||
|
||||
# Note: don't update the sha of previous versions because the install will take forever
|
||||
# instead, update the repo state in a later step
|
||||
|
||||
ARG SHA=2995107fa24cfd72b0a991e18271dcde148c2807
|
||||
ARG SHA=f49c08ea566385db339c6628f65c3a121033f67c
|
||||
RUN <<EOF
|
||||
cd stable-diffusion-webui
|
||||
git pull --rebase
|
||||
git reset --hard ${SHA}
|
||||
pip install --prefer-binary --no-cache-dir -r requirements_versions.txt
|
||||
pip install --prefer-binary --no-cache-dir -r requirements.txt
|
||||
pip install -r requirements_versions.txt
|
||||
EOF
|
||||
|
||||
RUN pip install --prefer-binary --no-cache-dir opencv-python-headless \
|
||||
RUN pip install opencv-python-headless \
|
||||
git+https://github.com/TencentARC/GFPGAN.git@8d2447a2d918f8eba5a4a01463fd48e45126a379 \
|
||||
git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1
|
||||
|
||||
git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1 \
|
||||
pyngrok
|
||||
|
||||
COPY --from=xformers /wheel.whl xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl
|
||||
RUN pip install xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl
|
||||
|
||||
COPY . /docker
|
||||
RUN <<EOF
|
||||
chmod +x /docker/mount.sh && python3 /docker/info.py ${ROOT}/modules/ui.py
|
||||
EOF
|
||||
|
||||
|
||||
ENV CLI_ARGS=""
|
||||
WORKDIR ${WORKDIR}
|
||||
EXPOSE 7860
|
||||
# run, -u to not buffer stdout / stderr
|
||||
CMD /docker/mount.sh && \
|
||||
python3 -u ../../webui.py --listen --port 7860 --hide-ui-dir-config --ckpt-dir ${ROOT}/models/Stable-diffusion ${CLI_ARGS}
|
||||
python3 -u ../../webui.py --listen --port 7860 --ckpt-dir ${ROOT}/models/Stable-diffusion --theme dark ${CLI_ARGS}
|
||||
|
||||
@@ -6,74 +6,5 @@
|
||||
"outdir_txt2img_grids": "/output/txt2img-grids",
|
||||
"outdir_img2img_grids": "/output/img2img-grids",
|
||||
"outdir_save": "/output/saved",
|
||||
"font": "DejaVuSans.ttf",
|
||||
"__WARNING__": "DON'T CHANGE ANYTHING BEFORE THIS",
|
||||
|
||||
"add_model_hash_to_info": false,
|
||||
"code_former_weight": 0.5,
|
||||
"directories_filename_pattern": "",
|
||||
"directories_max_prompt_words": 8,
|
||||
"enable_batch_seeds": true,
|
||||
"enable_emphasis": true,
|
||||
"enable_pnginfo": true,
|
||||
"enable_quantization": false,
|
||||
"ESRGAN_tile": 192,
|
||||
"ESRGAN_tile_overlap": 8,
|
||||
"export_for_4chan": true,
|
||||
"face_restoration_model": null,
|
||||
"face_restoration_unload": false,
|
||||
"filter_nsfw": false,
|
||||
"grid_extended_filename": false,
|
||||
"grid_format": "png",
|
||||
"grid_only_if_multiple": true,
|
||||
"grid_save": true,
|
||||
"grid_save_to_dirs": false,
|
||||
"img2img_color_correction": false,
|
||||
"img2img_fix_steps": false,
|
||||
"interrogate_clip_dict_limit": 1500,
|
||||
"interrogate_clip_max_length": 48,
|
||||
"interrogate_clip_min_length": 24,
|
||||
"interrogate_clip_num_beams": 1,
|
||||
"interrogate_keep_models_in_memory": false,
|
||||
"interrogate_use_builtin_artists": true,
|
||||
"jpeg_quality": 80,
|
||||
"js_modal_lightbox": true,
|
||||
"js_modal_lightbox_initialy_zoomed": true,
|
||||
"ldsr_post_down": 1,
|
||||
"ldsr_pre_down": 1,
|
||||
"ldsr_steps": 30,
|
||||
"memmon_poll_rate": 8,
|
||||
"multiple_tqdm": true,
|
||||
"n_rows": -1,
|
||||
"outdir_grids": "",
|
||||
"random_artist_categories": [],
|
||||
"realesrgan_enabled_models": [
|
||||
"Real-ESRGAN 4x plus",
|
||||
"Real-ESRGAN 4x plus anime 6B",
|
||||
"Real-ESRGAN 2x plus",
|
||||
"Real-ESRGAN AnimeVideo",
|
||||
"Real-ESRGAN General WDN x4x3",
|
||||
"Real-ESRGAN General x4x3"
|
||||
],
|
||||
"return_grid": true,
|
||||
"samples_filename_format": "",
|
||||
"samples_filename_pattern": "",
|
||||
"samples_format": "png",
|
||||
"samples_log_stdout": false,
|
||||
"samples_save": true,
|
||||
"save_images_before_color_correction": false,
|
||||
"save_images_before_face_restoration": false,
|
||||
"save_selected_only": false,
|
||||
"save_to_dirs": false,
|
||||
"save_to_dirs_prompt_len": 10,
|
||||
"save_txt": false,
|
||||
"sd_model_checkpoint": null,
|
||||
"show_progress_every_n_steps": 7,
|
||||
"show_progressbar": true,
|
||||
"SWIN_tile": 192,
|
||||
"SWIN_tile_overlap": 8,
|
||||
"upscale_at_full_resolution_padding": 16,
|
||||
"upscaler_for_hires_fix": null,
|
||||
"upscaler_for_img2img": null,
|
||||
"use_original_name_batch": false
|
||||
"font": "DejaVuSans.ttf"
|
||||
}
|
||||
|
||||
@@ -2,6 +2,14 @@
|
||||
|
||||
set -Eeuo pipefail
|
||||
|
||||
mkdir -p /data/config/auto/
|
||||
cp -n /docker/config.json /data/config/auto/config.json
|
||||
jq '. * input' /data/config/auto/config.json /docker/config.json | sponge /data/config/auto/config.json
|
||||
|
||||
if [ ! -f /data/config/auto/ui-config.json ]; then
|
||||
echo '{}' >/data/config/auto/ui-config.json
|
||||
fi
|
||||
|
||||
declare -A MOUNTS
|
||||
|
||||
MOUNTS["/root/.cache"]="/data/.cache"
|
||||
@@ -16,8 +24,11 @@ MOUNTS["${ROOT}/models/RealESRGAN"]="/data/RealESRGAN"
|
||||
MOUNTS["${ROOT}/models/SwinIR"]="/data/SwinIR"
|
||||
MOUNTS["${ROOT}/models/ScuNET"]="/data/ScuNET"
|
||||
MOUNTS["${ROOT}/models/LDSR"]="/data/LDSR"
|
||||
MOUNTS["${ROOT}/models/hypernetworks"]="/data/Hypernetworks"
|
||||
|
||||
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"
|
||||
|
||||
# extra hacks
|
||||
MOUNTS["${ROOT}/repositories/CodeFormer/weights/facelib"]="/data/.cache"
|
||||
@@ -26,8 +37,12 @@ for to_path in "${!MOUNTS[@]}"; do
|
||||
set -Eeuo pipefail
|
||||
from_path="${MOUNTS[${to_path}]}"
|
||||
rm -rf "${to_path}"
|
||||
mkdir -vp "$from_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
|
||||
|
||||
mkdir -p /output/saved /output/txt2img-images/ /output/img2img-images /output/extras-images/ /output/grids/ /output/txt2img-grids/ /output/img2img-grids/
|
||||
|
||||
@@ -4,13 +4,6 @@ set -Eeuo pipefail
|
||||
|
||||
mkdir -p /data/.cache /data/StableDiffusion /data/Codeformer /data/GFPGAN /data/ESRGAN /data/BSRGAN /data/RealESRGAN /data/SwinIR /data/LDSR /data/ScuNET /data/embeddings
|
||||
|
||||
cat <<EOF
|
||||
By using this software, you agree to the following licenses:
|
||||
https://github.com/CompVis/stable-diffusion/blob/main/LICENSE
|
||||
https://github.com/TencentARC/GFPGAN/blob/master/LICENSE
|
||||
https://github.com/xinntao/Real-ESRGAN/blob/master/LICENSE
|
||||
EOF
|
||||
|
||||
echo "Downloading, this might take a while..."
|
||||
|
||||
aria2c --input-file /docker/links.txt --dir /data --continue
|
||||
@@ -19,6 +12,18 @@ echo "Checking SHAs..."
|
||||
|
||||
parallel --will-cite -a /docker/checksums.sha256 "echo -n {} | sha256sum -c"
|
||||
|
||||
# fix potential permissions
|
||||
# TODO: need something better than this:
|
||||
# chmod -R 777 /data /output
|
||||
# aria2c already does hash check
|
||||
# cc6cb27103417325ff94f52b7a5d2dde45a7515b25c255d8e396c90014281516 /data/StableDiffusion/v1-5-pruned-emaonly.ckpt
|
||||
cat <<EOF
|
||||
By using this software, you agree to the following licenses:
|
||||
https://github.com/CompVis/stable-diffusion/blob/main/LICENSE
|
||||
https://github.com/AbdBarho/stable-diffusion-webui-docker/blob/master/LICENSE
|
||||
https://github.com/sd-webui/stable-diffusion-webui/blob/master/LICENSE
|
||||
https://github.com/invoke-ai/InvokeAI/blob/main/LICENSE
|
||||
https://github.com/cszn/BSRGAN/blob/main/LICENSE
|
||||
https://github.com/sczhou/CodeFormer/blob/master/LICENSE
|
||||
https://github.com/TencentARC/GFPGAN/blob/master/LICENSE
|
||||
https://github.com/xinntao/Real-ESRGAN/blob/master/LICENSE
|
||||
https://github.com/xinntao/ESRGAN/blob/master/LICENSE
|
||||
https://github.com/cszn/SCUNet/blob/main/LICENSE
|
||||
EOF
|
||||
|
||||
@@ -1,4 +1,12 @@
|
||||
https://www.googleapis.com/storage/v1/b/aai-blog-files/o/sd-v1-4.ckpt?alt=media
|
||||
magnet:?xt=urn:btih:2daef5b5f63a16a9af9169a529b1a773fc452637&dn=v1-5-pruned-emaonly.ckpt&tr=udp%3a%2f%2ftracker.opentrackr.org%3a1337%2fannounce&tr=udp%3a%2f%2f9.rarbg.com%3a2810%2fannounce&tr=udp%3a%2f%2ftracker.openbittorrent.com%3a6969%2fannounce&tr=udp%3a%2f%2fopentracker.i2p.rocks%3a6969%2fannounce&tr=https%3a%2f%2fopentracker.i2p.rocks%3a443%2fannounce&tr=http%3a%2f%2ftracker.openbittorrent.com%3a80%2fannounce&tr=udp%3a%2f%2ftracker.torrent.eu.org%3a451%2fannounce&tr=udp%3a%2f%2fopen.stealth.si%3a80%2fannounce&tr=udp%3a%2f%2fvibe.sleepyinternetfun.xyz%3a1738%2fannounce&tr=udp%3a%2f%2ftracker2.dler.org%3a80%2fannounce&tr=udp%3a%2f%2ftracker1.bt.moack.co.kr%3a80%2fannounce&tr=udp%3a%2f%2ftracker.zemoj.com%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.tiny-vps.com%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.theoks.net%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.publictracker.xyz%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.monitorit4.me%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.moeking.me%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.lelux.fi%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.dler.org%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.army%3a6969%2fannounce
|
||||
select-file=1
|
||||
index-out=1=StableDiffusion/v1-5-pruned-emaonly.ckpt
|
||||
follow-torrent=mem
|
||||
follow-metalink=mem
|
||||
seed-time=0
|
||||
# this is the only way aria2c won't fail if the file already exists
|
||||
check-integrity=true
|
||||
https://drive.yerf.org/wl/?id=EBfTrmcCCUAGaQBXVIj5lJmEhjoP1tgl&mode=grid&download=1
|
||||
out=StableDiffusion/model.ckpt
|
||||
https://github.com/TencentARC/GFPGAN/releases/download/v1.3.4/GFPGANv1.4.pth
|
||||
out=GFPGAN/GFPGANv1.4.pth
|
||||
|
||||
@@ -22,10 +22,7 @@ conda clean -a -y
|
||||
EOF
|
||||
|
||||
|
||||
# Note: don't update the sha of previous versions because the install will take forever
|
||||
# instead, update the repo state in a later step
|
||||
ARG BRANCH=master SHA=1a9c053cb7b6832695771db2555c0adc9b41e95f
|
||||
# ARG BRANCH=dev SHA=1e7bdfe3f38a6dd37fc230f440ea1b0db0937240
|
||||
ARG BRANCH=dev SHA=8d1e42b9c50c747d056b0a98f3c2eb7652fb73a7
|
||||
RUN <<EOF
|
||||
cd stable-diffusion
|
||||
git fetch
|
||||
@@ -35,16 +32,17 @@ conda env update --file environment.yaml -n base
|
||||
conda clean -a -y
|
||||
EOF
|
||||
|
||||
RUN pip install -U --no-cache-dir pyperclip
|
||||
|
||||
# add info
|
||||
COPY . /docker/
|
||||
RUN python /docker/info.py /stable-diffusion/frontend/frontend.py && chmod +x /docker/mount.sh
|
||||
RUN <<EOF
|
||||
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
|
||||
EOF
|
||||
|
||||
WORKDIR /stable-diffusion
|
||||
ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
|
||||
ENV PYTHONPATH="${PYTHONPATH}:${PWD}" STREAMLIT_SERVER_HEADLESS=true USE_STREAMLIT=0 CLI_ARGS=""
|
||||
EXPOSE 7860
|
||||
# run, -u to not buffer stdout / stderr
|
||||
CMD /docker/mount.sh && \
|
||||
python3 -u scripts/webui.py --outdir /output --ckpt /data/StableDiffusion/model.ckpt ${CLI_ARGS}
|
||||
# sed -i -- 's/8501/7860/g' .streamlit/config.toml && STREAMLIT_SERVER_HEADLESS=true python -u -m streamlit run scripts/webui_streamlit.py --theme.base dark
|
||||
|
||||
CMD /docker/mount.sh && /docker/run.sh
|
||||
|
||||
10
services/hlky/run.sh
Executable file
10
services/hlky/run.sh
Executable file
@@ -0,0 +1,10 @@
|
||||
#!/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,8 +1,9 @@
|
||||
general:
|
||||
outdir: /outputs
|
||||
default_model: "Stable Diffusion v1.4"
|
||||
default_model_path: /data/StableDiffusion/model.ckpt
|
||||
outdir_txt2img: /outputs/txt2img-samples
|
||||
outdir_img2img: /outputs/img2img-samples
|
||||
optimized: True
|
||||
version: 1.20.0
|
||||
outdir: /output
|
||||
default_model: "Stable Diffusion v1.5"
|
||||
default_model_path: /data/StableDiffusion/v1-5-pruned-emaonly.ckpt
|
||||
outdir_txt2img: /output/txt2img-samples
|
||||
outdir_img2img: /output/img2img-samples
|
||||
outdir_img2txt: /output/img2txt
|
||||
optimized_turbo: True
|
||||
|
||||
@@ -17,15 +17,14 @@ ENV PIP_EXISTS_ACTION=w
|
||||
RUN <<EOF
|
||||
git clone https://github.com/invoke-ai/InvokeAI.git stable-diffusion
|
||||
cd stable-diffusion
|
||||
git reset --hard 8a8be92eac17e0ef699528157596b2336bdee532
|
||||
sed -i -- 's/python=3.8.5/python=3.9/g' environment.yaml
|
||||
git reset --hard 79e79b78aaeedb49afcc795e0e00eebfdbedee96
|
||||
git config --global http.postBuffer 1048576000
|
||||
conda env update --file environment.yaml -n base
|
||||
conda env update --file environment.yml -n base
|
||||
conda clean -a -y
|
||||
EOF
|
||||
|
||||
|
||||
ARG BRANCH=development SHA=4f247a3672474bd9c46060bab6087dbf9e2531f3
|
||||
ARG BRANCH=development SHA=554445a985d970200095bbcb109273a49c462682
|
||||
RUN <<EOF
|
||||
cd stable-diffusion
|
||||
git fetch
|
||||
@@ -36,7 +35,7 @@ conda env update --file environment.yml -n base
|
||||
conda clean -a -y
|
||||
EOF
|
||||
|
||||
RUN pip uninstall opencv-python -y && pip install --prefer-binary --force-reinstall --no-cache-dir opencv-python-headless transformers==4.19.2
|
||||
RUN pip uninstall opencv-python -y && pip install --prefer-binary --force-reinstall --no-cache-dir opencv-python-headless
|
||||
|
||||
COPY . /docker/
|
||||
RUN <<EOF
|
||||
@@ -50,8 +49,4 @@ WORKDIR /stable-diffusion
|
||||
EXPOSE 7860
|
||||
|
||||
CMD /docker/mount.sh && \
|
||||
# python3 -u backend/server.py --host 0.0.0.0 --port 7860 --cors http://localhost:7860
|
||||
python3 -u scripts/dream.py --outdir /output --web --host 0.0.0.0 --port 7860 ${CLI_ARGS}
|
||||
# echo The lstein webUI is currently deactivated due to implementation limitations: \
|
||||
# https://github.com/invoke-ai/InvokeAI/blob/8c9f2ae705cf723d4a8a73c416e8d8bf2d746977/backend/modules/create_cmd_parser.py#L26 \
|
||||
# Once the path the output is fixed, the UI will be activated again
|
||||
python3 -u scripts/invoke.py --outdir /output --web --host 0.0.0.0 --port 7860 ${CLI_ARGS}
|
||||
|
||||
@@ -9,10 +9,11 @@ MOUNTS["/root/.cache"]=/data/.cache
|
||||
# ui specific
|
||||
MOUNTS["${PWD}/models/ldm/stable-diffusion-v1/model.ckpt"]=/data/StableDiffusion/model.ckpt
|
||||
MOUNTS["${PWD}/src/gfpgan/experiments/pretrained_models/GFPGANv1.4.pth"]=/data/GFPGAN/GFPGANv1.4.pth
|
||||
MOUNTS["${PWD}/ldm/dream/restoration/codeformer/weights"]=/data/CodeFormer
|
||||
MOUNTS["${PWD}/ldm/invoke/restoration/codeformer/weights"]=/data/Codeformer
|
||||
# hacks
|
||||
MOUNTS["/opt/conda/lib/python3.9/site-packages/facexlib/weights"]=/data/.cache
|
||||
MOUNTS["/opt/conda/lib/python3.9/site-packages/realesrgan/weights"]=/data/RealESRGAN
|
||||
MOUNTS["${PWD}/gfpgan/weights"]=/data/.cache
|
||||
|
||||
for to_path in "${!MOUNTS[@]}"; do
|
||||
set -Eeuo pipefail
|
||||
|
||||
Reference in New Issue
Block a user