mirror of
				https://github.com/AbdBarho/stable-diffusion-webui-docker.git
				synced 2025-10-31 02:03:32 -04:00 
			
		
		
		
	Compare commits
	
		
			44 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|  | ab651fe0d7 | ||
|  | f76f8d4671 | ||
|  | e32a48f42a | ||
|  | 76989b39a6 | ||
|  | 4d9fc381bb | ||
|  | bcee253fe0 | ||
|  | 499143009a | ||
|  | c614625f04 | ||
|  | ccd6e238b2 | ||
|  | 829864af9b | ||
|  | ccc7306f48 | ||
|  | 082876aab3 | ||
|  | ae834cb764 | ||
|  | 5f6d9fbb03 | ||
|  | d4da252343 | ||
|  | 5af482ed8c | ||
|  | ce4e190f8f | ||
|  | bae3590980 | ||
|  | 1588d1eecf | ||
|  | 9cbd58b3f4 | ||
|  | 089fc524d8 | ||
|  | 0d8b7d4ac8 | ||
|  | 561664ea6e | ||
|  | 77c2b2d217 | ||
|  | 6c0c610f27 | ||
|  | dc730b7f6b | ||
|  | 15952906a1 | ||
|  | 4aaf38970a | ||
|  | 61bd38dfe4 | ||
|  | bec4997639 | ||
|  | 44903ca3aa | ||
|  | af409daa0e | ||
|  | 19f9402076 | ||
|  | eef83a318c | ||
|  | bc8ec0fe55 | ||
|  | f5e9997a55 | ||
|  | fec97dc9e1 | ||
|  | 134130a3b2 | ||
|  | 2ecfb0fe59 | ||
|  | 89d8a17064 | ||
|  | 05829c5ae3 | ||
|  | 66c0658255 | ||
|  | b3fa72a614 | ||
|  | 3efd62aac2 | 
							
								
								
									
										9
									
								
								.editorconfig
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								.editorconfig
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,9 @@ | ||||
| root = true | ||||
|  | ||||
| [*] | ||||
| end_of_line = lf | ||||
| indent_style = space | ||||
| indent_size = 2 | ||||
| charset = utf-8 | ||||
| insert_final_newline = true | ||||
| trim_trailing_whitespace = true | ||||
							
								
								
									
										1
									
								
								.gitattributes
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								.gitattributes
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1 @@ | ||||
| * text=auto eol=lf | ||||
							
								
								
									
										29
									
								
								.github/ISSUE_TEMPLATE/bug.md
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								.github/ISSUE_TEMPLATE/bug.md
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,29 @@ | ||||
| --- | ||||
| name: Bug | ||||
| about: Report a bug | ||||
| title: '' | ||||
| labels: bug | ||||
| 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) and in [the issues in the WebUI repo](https://github.com/hlky/stable-diffusion-webui)** | ||||
|  | ||||
|  | ||||
|  | ||||
| **Describe the bug** | ||||
|  | ||||
|  | ||||
| **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] | ||||
|  - GPU: [Nvidia 1660 / No GPU] | ||||
|  - Version [e.g. 22] | ||||
|  | ||||
| **Additional context** | ||||
| Any other context about the problem here. If applicable, add screenshots to help explain your problem. | ||||
							
								
								
									
										14
									
								
								.github/workflows/docker.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								.github/workflows/docker.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,14 @@ | ||||
| name: Build Images | ||||
|  | ||||
| on: [push] | ||||
|  | ||||
| jobs: | ||||
|   build_all: | ||||
|     runs-on: ubuntu-latest | ||||
|     name: All | ||||
|     steps: | ||||
|       - uses: actions/checkout@v3 | ||||
|       # better caching? | ||||
|       - run: docker compose --profile auto build --progress plain | ||||
|       - run: docker compose --profile hlky build --progress plain | ||||
|       - run: docker compose --profile lstein build --progress plain | ||||
							
								
								
									
										22
									
								
								.github/workflows/executable.yml1
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								.github/workflows/executable.yml1
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,22 @@ | ||||
| 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} | ||||
							
								
								
									
										22
									
								
								.vscode/launch.json
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								.vscode/launch.json
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,22 @@ | ||||
| { | ||||
|   // Use IntelliSense to learn about possible attributes. | ||||
|   // Hover to view descriptions of existing attributes. | ||||
|   // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387 | ||||
|   "version": "0.2.0", | ||||
|   "configurations": [ | ||||
|     { | ||||
|       "name": "WebUI", | ||||
|       "type": "python", | ||||
|       "request": "launch", | ||||
|       "program": "${file}", | ||||
|       "cwd": "/stable-diffusion", | ||||
|       "args": ["--ckpt", "${workspaceFolder}/models/model.ckpt", "--gfpgan-dir", "${workspaceFolder}/models/", "--extra-models-cpu"], | ||||
|       "env": { | ||||
|         "TRANSFORMERS_CACHE":"${workspaceFolder}/cache/transformers", | ||||
|         "TORCH_HOME":"${workspaceFolder}/cache/torch" | ||||
|       }, | ||||
|       "console": "integratedTerminal", | ||||
|       "justMyCode": false | ||||
|     } | ||||
|   ] | ||||
| } | ||||
| @@ -1,45 +0,0 @@ | ||||
| # syntax=docker/dockerfile:1 | ||||
|  | ||||
| FROM continuumio/miniconda3:4.12.0 | ||||
|  | ||||
|  | ||||
| RUN conda install python=3.8.5 && conda clean -a -y | ||||
| RUN conda install pytorch==1.11.0 torchvision==0.12.0 cudatoolkit=11.3 -c pytorch && conda clean -a -y | ||||
| RUN git clone https://github.com/hlky/stable-diffusion.git && cd stable-diffusion && git reset --hard 554bd068e6f2f6bc55449a67fe017ddd77090f28 | ||||
| RUN conda env update --file stable-diffusion/environment.yaml --name base && conda clean -a -y | ||||
|  | ||||
| # fonts for generating the grid | ||||
| RUN apt-get update && apt install fonts-dejavu-core && apt-get clean | ||||
|  | ||||
| # Note: don't update the sha of previous versions because the install will take forever | ||||
| # instead, update the repo state in a later step | ||||
| RUN cd stable-diffusion && git pull && git reset --hard 6b442c05296dd350fcf7f752e17c4b4bb015a03b && \ | ||||
|   conda env update --file environment.yaml --name base && conda clean -a -y | ||||
|  | ||||
| # download latest UI version, update the sha below in case you want some other version | ||||
| RUN git clone https://github.com/hlky/stable-diffusion-webui.git && cd stable-diffusion-webui && \ | ||||
|   git reset --hard 33890563d237889aca7cbc4f04ae42b885aee07e && \ | ||||
|   cp -t /stable-diffusion/scripts/ txt2img.yaml webui.py webui.yaml webui_playground.py && \ | ||||
|   cd / && rm -rf stable-diffusion-webui | ||||
|  | ||||
|  | ||||
| # For testing different versions: | ||||
| # RUN git clone https://github.com/AbdBarho/stable-diffusion-webui.git && cd stable-diffusion-webui && \ | ||||
| #   git checkout respect-cli-params &&\ | ||||
| #   cp -t /stable-diffusion/scripts/ txt2img.yaml webui.py webui.yaml webui_playground.py && \ | ||||
| #   cd / && rm -rf stable-diffusion-webui | ||||
|  | ||||
|  | ||||
| WORKDIR /stable-diffusion | ||||
| ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS="" | ||||
|  | ||||
| EXPOSE 7860 | ||||
| CMD \ | ||||
|   ln -sf /models/model.ckpt /stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt && \ | ||||
|   ln -sf /models/GFPGANv1.3.pth   /stable-diffusion/src/gfpgan/experiments/pretrained_models/GFPGANv1.3.pth && \ | ||||
|   ln -sf /models/RealESRGAN_x4plus.pth   /stable-diffusion/src/realesrgan/experiments/pretrained_models/RealESRGAN_x4plus.pth && \ | ||||
|   # force facexlib cache | ||||
|   mkdir -p /cache/weights/ && rm -rf /opt/conda/lib/python3.8/site-packages/facexlib/weights && \ | ||||
|   ln -sf  /cache/weights/ /opt/conda/lib/python3.8/site-packages/facexlib/ && \ | ||||
|   # run, -u to not buffer stdout / stderr | ||||
|   python3 -u scripts/webui.py --outdir /output --save-metadata --ckpt /models/model.ckpt ${CLI_ARGS} | ||||
							
								
								
									
										3
									
								
								cache/.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										3
									
								
								cache/.gitignore
									
									
									
									
										vendored
									
									
								
							| @@ -1,3 +1,4 @@ | ||||
| /torch | ||||
| /transformers | ||||
| /weights | ||||
| /weights | ||||
| /models | ||||
|   | ||||
| @@ -1,17 +1,11 @@ | ||||
| version: '3.9' | ||||
|  | ||||
| services: | ||||
|   model: | ||||
|     build: ./build/ | ||||
|     restart: on-failure | ||||
| x-base_service: &base_service | ||||
|     ports: | ||||
|       - "7860:7860" | ||||
|     volumes: | ||||
|       - ./cache:/cache | ||||
|       - ./output:/output | ||||
|       - ./models:/models | ||||
|     environment: | ||||
|       - CLI_ARGS=--extra-models-cpu | ||||
|       - &v1 ./cache:/cache | ||||
|       - &v2 ./output:/output | ||||
|     deploy: | ||||
|       resources: | ||||
|         reservations: | ||||
| @@ -19,3 +13,42 @@ services: | ||||
|               - driver: nvidia | ||||
|                 device_ids: ['0'] | ||||
|                 capabilities: [gpu] | ||||
|  | ||||
| name: webui-docker | ||||
|  | ||||
| services: | ||||
|   download: | ||||
|     build: ./services/download/ | ||||
|     profiles: ["download"] | ||||
|     volumes: | ||||
|       - *v1 | ||||
|  | ||||
|   hlky: | ||||
|     <<: *base_service | ||||
|     profiles: ["hlky"] | ||||
|     build: ./services/hlky/ | ||||
|     environment: | ||||
|       - CLI_ARGS=--optimized-turbo | ||||
|  | ||||
|   automatic1111: &automatic | ||||
|     <<: *base_service | ||||
|     profiles: ["auto"] | ||||
|     build: ./services/AUTOMATIC1111 | ||||
|     volumes: | ||||
|       - *v1 | ||||
|       - *v2 | ||||
|       - ./services/AUTOMATIC1111/config.json:/docker/config.json | ||||
|     environment: | ||||
|       - CLI_ARGS=--medvram --opt-split-attention | ||||
|  | ||||
|   automatic1111-cpu: | ||||
|     <<: *automatic | ||||
|     profiles: ["auto-cpu"] | ||||
|     deploy: {} | ||||
|     environment: | ||||
|       - CLI_ARGS=--no-half --precision full | ||||
|  | ||||
|   lstein: | ||||
|     <<: *base_service | ||||
|     profiles: ["lstein"] | ||||
|     build: ./services/lstein/ | ||||
|   | ||||
							
								
								
									
										4
									
								
								models/.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										4
									
								
								models/.gitignore
									
									
									
									
										vendored
									
									
								
							| @@ -1,4 +0,0 @@ | ||||
| /model.ckpt | ||||
| /GFPGANv1.3.pth | ||||
| /RealESRGAN_x4plus.pth | ||||
| /RealESRGAN_x4plus_anime_6B.pth | ||||
							
								
								
									
										5
									
								
								scripts/chmod.sh
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										5
									
								
								scripts/chmod.sh
									
									
									
									
									
										Executable file
									
								
							| @@ -0,0 +1,5 @@ | ||||
| #!/bin/bash | ||||
|  | ||||
| set -Eeuo pipefail | ||||
|  | ||||
| find . -name "*.sh" -exec git update-index --chmod=+x {} \; | ||||
							
								
								
									
										62
									
								
								services/AUTOMATIC1111/Dockerfile
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										62
									
								
								services/AUTOMATIC1111/Dockerfile
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,62 @@ | ||||
| # syntax=docker/dockerfile:1 | ||||
|  | ||||
| FROM alpine/git:2.36.2 as download | ||||
| RUN <<EOF | ||||
| # who knows | ||||
| git config --global http.postBuffer 1048576000 | ||||
| git clone https://github.com/sczhou/CodeFormer.git repositories/CodeFormer | ||||
| git clone https://github.com/CompVis/stable-diffusion.git repositories/stable-diffusion | ||||
| git clone https://github.com/CompVis/taming-transformers.git repositories/taming-transformers | ||||
| rm -rf repositories/taming-transformers/data repositories/taming-transformers/assets | ||||
| EOF | ||||
|  | ||||
|  | ||||
| FROM continuumio/miniconda3:4.12.0 | ||||
|  | ||||
| SHELL ["/bin/bash", "-ceuxo", "pipefail"] | ||||
|  | ||||
| ENV DEBIAN_FRONTEND=noninteractive | ||||
|  | ||||
| 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 apt-get update && apt install fonts-dejavu-core rsync -y && apt-get clean | ||||
|  | ||||
|  | ||||
| RUN <<EOF | ||||
| git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git | ||||
| cd stable-diffusion-webui | ||||
| git reset --hard 13eec4f3d4081fdc43883c5ef02e471a2b6c7212 | ||||
| conda env update --file environment-wsl2.yaml -n base | ||||
| conda clean -a -y | ||||
| pip install --prefer-binary --no-cache-dir -r requirements.txt | ||||
| EOF | ||||
|  | ||||
| ENV ROOT=/stable-diffusion-webui \ | ||||
|   WORKDIR=/stable-diffusion-webui/repositories/stable-diffusion | ||||
|  | ||||
| COPY --from=download /git/ ${ROOT} | ||||
| RUN pip install --prefer-binary --no-cache-dir -r ${ROOT}/repositories/CodeFormer/requirements.txt | ||||
|  | ||||
| # 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=b5d1af11b7dc718d4d91d379c75e46f4bd2e2fe6 | ||||
| RUN <<EOF | ||||
| cd stable-diffusion-webui | ||||
| git pull --rebase | ||||
| git reset --hard ${SHA} | ||||
| pip install --prefer-binary --no-cache-dir -r requirements.txt | ||||
| EOF | ||||
|  | ||||
| RUN pip install --prefer-binary -U --no-cache-dir opencv-python-headless markupsafe==2.0.1 | ||||
|  | ||||
| ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS="" | ||||
|  | ||||
| COPY . /docker | ||||
| RUN chmod +x /docker/mount.sh && python3 /docker/info.py ${ROOT}/modules/ui.py | ||||
|  | ||||
|  | ||||
| WORKDIR ${WORKDIR} | ||||
| EXPOSE 7860 | ||||
| # run, -u to not buffer stdout / stderr | ||||
| CMD /docker/mount.sh && python3 -u ../../webui.py --listen --port 7860 ${CLI_ARGS} | ||||
							
								
								
									
										14
									
								
								services/AUTOMATIC1111/README.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								services/AUTOMATIC1111/README.md
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,14 @@ | ||||
| # WebUI for AUTOMATIC1111 | ||||
|  | ||||
| The WebUI of [AUTOMATIC1111/stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui) as docker container! | ||||
|  | ||||
| ## Setup | ||||
|  | ||||
| Clone this repo, download the `model.ckpt` and `GFPGANv1.3.pth` and put into the `models` folder as mentioned in [the main README](../README.md), then run | ||||
|  | ||||
| ``` | ||||
| cd AUTOMATIC1111 | ||||
| docker compose up --build | ||||
| ``` | ||||
|  | ||||
| You can change the cli parameters in `AUTOMATIC1111/docker-compose.yml`. The full list of cil parameters can be found [here](https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/master/modules/shared.py) | ||||
							
								
								
									
										1
									
								
								services/AUTOMATIC1111/config.json
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/AUTOMATIC1111/config.json
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1 @@ | ||||
| {"outdir_samples": "/output", "outdir_txt2img_samples": "/output/txt2img-images", "outdir_img2img_samples": "/output/img2img-images", "outdir_extras_samples": "/output/extras-images", "outdir_txt2img_grids": "/output/txt2img-grids", "outdir_img2img_grids": "/output/img2img-grids", "outdir_save": "/output/saved", "__WARNING__": "DON'T CHANGE ANYTHING BEFORE THIS", "outdir_grids": "", "save_to_dirs": false, "save_to_dirs_prompt_len": 10, "samples_save": true, "samples_format": "png", "grid_save": true, "return_grid": true, "grid_format": "png", "grid_extended_filename": false, "grid_only_if_multiple": true, "n_rows": -1, "jpeg_quality": 80, "export_for_4chan": true, "enable_pnginfo": true, "font": "DejaVuSans.ttf", "enable_emphasis": true, "save_txt": false, "ESRGAN_tile": 192, "ESRGAN_tile_overlap": 8, "random_artist_categories": [], "upscale_at_full_resolution_padding": 16, "show_progressbar": true, "show_progress_every_n_steps": 5, "multiple_tqdm": true, "face_restoration_model": "CodeFormer", "code_former_weight": 0.5, "grid_save_to_dirs": false} | ||||
							
								
								
									
										14
									
								
								services/AUTOMATIC1111/info.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								services/AUTOMATIC1111/info.py
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,14 @@ | ||||
| import sys | ||||
| from pathlib import Path | ||||
|  | ||||
| file = Path(sys.argv[1]) | ||||
| file.write_text( | ||||
|   file.read_text()\ | ||||
|   .replace('    return demo', """ | ||||
|     with demo: | ||||
|         gr.Markdown( | ||||
|           'Created by [AUTOMATIC1111 / stable-diffusion-webui-docker](https://github.com/AbdBarho/stable-diffusion-webui-docker/tree/master/AUTOMATIC1111)' | ||||
|         ) | ||||
|     return demo | ||||
| """, 1) | ||||
| ) | ||||
							
								
								
									
										36
									
								
								services/AUTOMATIC1111/mount.sh
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										36
									
								
								services/AUTOMATIC1111/mount.sh
									
									
									
									
									
										Executable file
									
								
							| @@ -0,0 +1,36 @@ | ||||
| #!/bin/bash | ||||
|  | ||||
| set -e | ||||
|  | ||||
| declare -A MODELS | ||||
|  | ||||
| MODELS["${WORKDIR}/models/ldm/stable-diffusion-v1/model.ckpt"]=model.ckpt | ||||
| MODELS["${ROOT}/GFPGANv1.3.pth"]=GFPGANv1.3.pth | ||||
|  | ||||
| MODELS_DIR=/cache/models | ||||
|  | ||||
| for path in "${!MODELS[@]}"; do | ||||
|   name=${MODELS[$path]} | ||||
|   base=$(dirname "${path}") | ||||
|   from_path="${MODELS_DIR}/${name}" | ||||
|   if test -f "${from_path}"; then | ||||
|     mkdir -p "${base}" && ln -sf "${from_path}" "${path}" && echo "Mounted ${name}" | ||||
|   else | ||||
|     echo "Skipping ${name}" | ||||
|   fi | ||||
| done | ||||
|  | ||||
| # force realesrgan cache | ||||
| rm -rf /opt/conda/lib/python3.8/site-packages/realesrgan/weights | ||||
| ln -s -T "${MODELS_DIR}" /opt/conda/lib/python3.8/site-packages/realesrgan/weights | ||||
|  | ||||
| # force facexlib cache | ||||
| mkdir -p /cache/weights/ ${WORKDIR}/gfpgan/ | ||||
| ln -sf /cache/weights/ ${WORKDIR}/gfpgan/ | ||||
| # code former cache | ||||
| rm -rf ${ROOT}/repositories/CodeFormer/weights/CodeFormer ${ROOT}/repositories/CodeFormer/weights/facelib | ||||
| ln -sf -T /cache/weights ${ROOT}/repositories/CodeFormer/weights/CodeFormer | ||||
| ln -sf -T /cache/weights ${ROOT}/repositories/CodeFormer/weights/facelib | ||||
|  | ||||
| # mount config | ||||
| ln -sf /docker/config.json ${WORKDIR}/config.json | ||||
							
								
								
									
										6
									
								
								services/download/Dockerfile
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										6
									
								
								services/download/Dockerfile
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,6 @@ | ||||
| FROM bash:alpine3.15 | ||||
|  | ||||
| RUN apk add parallel | ||||
| COPY . /docker | ||||
| RUN chmod +x /docker/download.sh | ||||
| ENTRYPOINT ["/docker/download.sh"] | ||||
							
								
								
									
										6
									
								
								services/download/checksums.sha256
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										6
									
								
								services/download/checksums.sha256
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,6 @@ | ||||
| fe4efff1e174c627256e44ec2991ba279b3816e364b49f9be2abc0b3ff3f8556  /cache/models/model.ckpt | ||||
| c953a88f2727c85c3d9ae72e2bd4846bbaf59fe6972ad94130e23e7017524a70  /cache/models/GFPGANv1.3.pth | ||||
| 4fa0d38905f75ac06eb49a7951b426670021be3018265fd191d2125df9d682f1  /cache/models/RealESRGAN_x4plus.pth | ||||
| f872d837d3c90ed2e05227bed711af5671a6fd1c9f7d7e91c911a61f155e99da  /cache/models/RealESRGAN_x4plus_anime_6B.pth | ||||
| c209caecac2f97b4bb8f4d726b70ac2ac9b35904b7fc99801e1f5e61f9210c13  /cache/models/LDSR.ckpt | ||||
| 9d6ad53c5dafeb07200fb712db14b813b527edd262bc80ea136777bdb41be2ba  /cache/models/LDSR.yaml | ||||
							
								
								
									
										33
									
								
								services/download/download.sh
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										33
									
								
								services/download/download.sh
									
									
									
									
									
										Executable file
									
								
							| @@ -0,0 +1,33 @@ | ||||
| #!/usr/bin/env bash | ||||
|  | ||||
| set -Eeuo pipefail | ||||
|  | ||||
| # [[ "$(sha256sum -b $file | head -c 64)" == "$sha" ]] | ||||
|  | ||||
| declare -A MODELS | ||||
|  | ||||
| MODELS['model.ckpt']='https://www.googleapis.com/storage/v1/b/aai-blog-files/o/sd-v1-4.ckpt?alt=media' | ||||
| MODELS['GFPGANv1.3.pth']='https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth' | ||||
| MODELS['RealESRGAN_x4plus.pth']='https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth' | ||||
| MODELS['RealESRGAN_x4plus_anime_6B.pth']='https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth' | ||||
| MODELS['LDSR.yaml']='https://heibox.uni-heidelberg.de/f/31a76b13ea27482981b4/?dl=1' | ||||
| MODELS['LDSR.ckpt']='https://heibox.uni-heidelberg.de/f/578df07c8fc04ffbadf3/?dl=1' | ||||
|  | ||||
| echo "Downloading..." | ||||
|  | ||||
| for file in "${!MODELS[@]}"; do | ||||
|   url=${MODELS[$file]} | ||||
|   full_path="/cache/models/$file" | ||||
|  | ||||
|   if [[ -f "$full_path" ]]; then | ||||
|     echo "- $file exists" | ||||
|     continue | ||||
|   fi | ||||
|  | ||||
|   mkdir -p $(dirname $full_path) | ||||
|   wget --tries=10 -c -O $full_path $url | ||||
| done | ||||
|  | ||||
| echo "Checking SHAs..." | ||||
|  | ||||
| time parallel --will-cite -a /docker/checksums.sha256 "echo -n {} | sha256sum -c" | ||||
							
								
								
									
										59
									
								
								services/hlky/Dockerfile
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										59
									
								
								services/hlky/Dockerfile
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,59 @@ | ||||
| # syntax=docker/dockerfile:1 | ||||
|  | ||||
| FROM continuumio/miniconda3:4.12.0 | ||||
|  | ||||
| SHELL ["/bin/bash", "-ceuxo", "pipefail"] | ||||
|  | ||||
| ENV DEBIAN_FRONTEND=noninteractive | ||||
|  | ||||
| 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 apt-get update && apt install fonts-dejavu-core rsync -y && apt-get clean | ||||
|  | ||||
|  | ||||
| RUN <<EOF | ||||
| git clone https://github.com/sd-webui/stable-diffusion-webui.git stable-diffusion | ||||
| cd stable-diffusion | ||||
| git reset --hard 2b1ac8daf7ea82c6c56eabab7e80ec1c33106a98 | ||||
| conda env update --file environment.yaml -n base | ||||
| conda clean -a -y | ||||
| EOF | ||||
|  | ||||
| # new dependency, should be added to the environment.yaml | ||||
| RUN pip install -U --no-cache-dir pyperclip | ||||
|  | ||||
| # 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 | ||||
| ARG SHA=2236e8b5854092054e2c30edc559006ace53bf96 | ||||
| RUN <<EOF | ||||
| cd stable-diffusion | ||||
| git fetch | ||||
| git checkout ${BRANCH} | ||||
| git reset --hard ${SHA} | ||||
| conda env update --file environment.yaml -n base | ||||
| conda clean -a -y | ||||
| EOF | ||||
|  | ||||
| # Latent diffusion | ||||
| RUN <<EOF | ||||
| git clone https://github.com/Hafiidz/latent-diffusion.git | ||||
| cd latent-diffusion | ||||
| git reset --hard e1a84a89fcbb49881546cf2acf1e7e250923dba0 | ||||
| # hacks all the way down | ||||
| mv ldm ldm_latent && | ||||
| sed -i -- 's/from ldm/from ldm_latent/g' *.py | ||||
| # dont forget to update the yaml!! | ||||
| EOF | ||||
|  | ||||
|  | ||||
| # add info | ||||
| COPY . /docker/ | ||||
| RUN python /docker/info.py /stable-diffusion/frontend/frontend.py && chmod +x /docker/mount.sh | ||||
|  | ||||
| WORKDIR /stable-diffusion | ||||
| ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS="" | ||||
| EXPOSE 7860 | ||||
| # run, -u to not buffer stdout / stderr | ||||
| CMD /docker/mount.sh && python3 -u scripts/webui.py --outdir /output --ckpt /cache/models/model.ckpt --ldsr-dir /latent-diffusion ${CLI_ARGS} | ||||
							
								
								
									
										13
									
								
								services/hlky/info.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										13
									
								
								services/hlky/info.py
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,13 @@ | ||||
| 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) | ||||
| ) | ||||
							
								
								
									
										35
									
								
								services/hlky/mount.sh
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										35
									
								
								services/hlky/mount.sh
									
									
									
									
									
										Executable file
									
								
							| @@ -0,0 +1,35 @@ | ||||
| #!/bin/bash | ||||
|  | ||||
| set -e | ||||
|  | ||||
| declare -A MODELS | ||||
|  | ||||
| ROOT=/stable-diffusion/src | ||||
|  | ||||
| MODELS["${ROOT}/gfpgan/experiments/pretrained_models/GFPGANv1.3.pth"]=GFPGANv1.3.pth | ||||
| MODELS["${ROOT}/realesrgan/experiments/pretrained_models/RealESRGAN_x4plus.pth"]=RealESRGAN_x4plus.pth | ||||
| MODELS["${ROOT}/realesrgan/experiments/pretrained_models/RealESRGAN_x4plus_anime_6B.pth"]=RealESRGAN_x4plus_anime_6B.pth | ||||
| MODELS["/latent-diffusion/experiments/pretrained_models/model.ckpt"]=LDSR.ckpt | ||||
| # MODELS["/latent-diffusion/experiments/pretrained_models/project.yaml"]=LDSR.yaml | ||||
|  | ||||
| MODELS_DIR=/cache/models | ||||
|  | ||||
| for path in "${!MODELS[@]}"; do | ||||
|   name=${MODELS[$path]} | ||||
|   base=$(dirname "${path}") | ||||
|   from_path="${MODELS_DIR}/${name}" | ||||
|   if test -f "${from_path}"; then | ||||
|     mkdir -p "${base}" && ln -sf "${from_path}" "${path}" && echo "Mounted ${name}" | ||||
|   else | ||||
|     echo "Skipping ${name}" | ||||
|   fi | ||||
| done | ||||
|  | ||||
| # hack for latent-diffusion | ||||
| if test -f "${MODELS_DIR}/LDSR.yaml"; then | ||||
|   sed 's/ldm\./ldm_latent\./g' "${MODELS_DIR}/LDSR.yaml" >/latent-diffusion/experiments/pretrained_models/project.yaml | ||||
| fi | ||||
|  | ||||
| # force facexlib cache | ||||
| mkdir -p /cache/weights/ /stable-diffusion/gfpgan/ | ||||
| ln -sf /cache/weights/ /stable-diffusion/gfpgan/ | ||||
							
								
								
									
										31
									
								
								services/lstein/Dockerfile
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								services/lstein/Dockerfile
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,31 @@ | ||||
| # syntax=docker/dockerfile:1 | ||||
|  | ||||
| FROM continuumio/miniconda3:4.12.0 | ||||
|  | ||||
| SHELL ["/bin/bash", "-ceuxo", "pipefail"] | ||||
|  | ||||
| ENV DEBIAN_FRONTEND=noninteractive | ||||
|  | ||||
| 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 apt-get update && apt install fonts-dejavu-core rsync -y && apt-get clean | ||||
|  | ||||
|  | ||||
| RUN <<EOF | ||||
| git clone https://github.com/lstein/stable-diffusion.git | ||||
| cd stable-diffusion | ||||
| git reset --hard 751283a2de81bee4bb571fbabe4adb19f1d85b97 | ||||
| conda env update --file environment.yaml -n base | ||||
| conda clean -a -y | ||||
| EOF | ||||
|  | ||||
| ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS="" | ||||
|  | ||||
| WORKDIR /stable-diffusion | ||||
|  | ||||
| EXPOSE 7860 | ||||
| # run, -u to not buffer stdout / stderr | ||||
| CMD mkdir -p /stable-diffusion/models/ldm/stable-diffusion-v1/ && \ | ||||
|   ln -sf /cache/models/model.ckpt /stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt && \ | ||||
|   python3 -u scripts/dream.py --outdir /output --web --host 0.0.0.0 --port 7860 ${CLI_ARGS} | ||||
							
								
								
									
										14
									
								
								services/lstein/README.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								services/lstein/README.md
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,14 @@ | ||||
| # WebUI for lstein | ||||
|  | ||||
| The WebUI of [lstein/stable-diffusion](https://github.com/lstein/stable-diffusion) as docker container! | ||||
|  | ||||
| Although it is a simple UI, the project has a lot of potential. | ||||
|  | ||||
| ## Setup | ||||
|  | ||||
| Clone this repo, download the `model.ckpt` and put into the `models` folder as mentioned in [the main README](../README.md), then run | ||||
|  | ||||
| ``` | ||||
| cd lstein | ||||
| docker compose up --build | ||||
| ``` | ||||
		Reference in New Issue
	
	Block a user