mirror of
				https://github.com/AbdBarho/stable-diffusion-webui-docker.git
				synced 2025-10-31 02:03:32 -04:00 
			
		
		
		
	Compare commits
	
		
			48 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|  | 5eae2076ce | ||
|  | 725e1f39ba | ||
|  | 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 | ||
|  | b5537dbc78 | ||
|  | 76bd1bf3cb | 
							
								
								
									
										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. | ||||||
							
								
								
									
										15
									
								
								.github/workflows/docker.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								.github/workflows/docker.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,15 @@ | |||||||
|  | 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 | ||||||
|  |       - run: docker compose --profile download 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,31 +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 |  | ||||||
| RUN apt-get update && apt install fonts-dejavu-core && apt-get clean |  | ||||||
|  |  | ||||||
| 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 |  | ||||||
|  |  | ||||||
|  |  | ||||||
| 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} |  | ||||||
							
								
								
									
										1
									
								
								cache/.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								cache/.gitignore
									
									
									
									
										vendored
									
									
								
							| @@ -1,3 +1,4 @@ | |||||||
| /torch | /torch | ||||||
| /transformers | /transformers | ||||||
| /weights | /weights | ||||||
|  | /models | ||||||
|   | |||||||
| @@ -1,17 +1,11 @@ | |||||||
| version: '3.9' | version: '3.9' | ||||||
|  |  | ||||||
| services: | x-base_service: &base_service | ||||||
|   model: |  | ||||||
|     build: ./build/ |  | ||||||
|     restart: on-failure |  | ||||||
|     ports: |     ports: | ||||||
|       - "7860:7860" |       - "7860:7860" | ||||||
|     volumes: |     volumes: | ||||||
|       - ./cache:/cache |       - &v1 ./cache:/cache | ||||||
|       - ./output:/output |       - &v2 ./output:/output | ||||||
|       - ./models:/models |  | ||||||
|     environment: |  | ||||||
|       - CLI_ARGS=--extra-models-cpu |  | ||||||
|     deploy: |     deploy: | ||||||
|       resources: |       resources: | ||||||
|         reservations: |         reservations: | ||||||
| @@ -19,3 +13,42 @@ services: | |||||||
|               - driver: nvidia |               - driver: nvidia | ||||||
|                 device_ids: ['0'] |                 device_ids: ['0'] | ||||||
|                 capabilities: [gpu] |                 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:/stable-diffusion-webui/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 {} \; | ||||||
							
								
								
									
										63
									
								
								services/AUTOMATIC1111/Dockerfile
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										63
									
								
								services/AUTOMATIC1111/Dockerfile
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,63 @@ | |||||||
|  | # syntax=docker/dockerfile:1 | ||||||
|  |  | ||||||
|  | FROM alpine/git:2.36.2 as download | ||||||
|  | RUN <<EOF | ||||||
|  | # because taming-transformers is huge | ||||||
|  | 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/salesforce/BLIP.git repositories/BLIP | ||||||
|  | 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=744ac1f89a075be4535146279feef800214c35a8 | ||||||
|  | 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 gfpgan==1.3.5 | ||||||
|  |  | ||||||
|  | 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 --hide-ui-dir-config ${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) | ||||||
							
								
								
									
										48
									
								
								services/AUTOMATIC1111/config.json
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										48
									
								
								services/AUTOMATIC1111/config.json
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,48 @@ | |||||||
|  | { | ||||||
|  |   "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", | ||||||
|  |   "font": "DejaVuSans.ttf", | ||||||
|  |   "__WARNING__": "DON'T CHANGE ANYTHING BEFORE THIS", | ||||||
|  |  | ||||||
|  |   "samples_filename_format": "", | ||||||
|  |   "outdir_grids": "", | ||||||
|  |   "save_to_dirs": false, | ||||||
|  |   "grid_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, | ||||||
|  |   "add_model_hash_to_info": false, | ||||||
|  |   "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": 7, | ||||||
|  |   "multiple_tqdm": true, | ||||||
|  |   "face_restoration_model": null, | ||||||
|  |   "code_former_weight": 0.5, | ||||||
|  |   "save_images_before_face_restoration": false, | ||||||
|  |   "face_restoration_unload": false, | ||||||
|  |   "interrogate_keep_models_in_memory": false, | ||||||
|  |   "interrogate_use_builtin_artists": true, | ||||||
|  |   "interrogate_clip_num_beams": 1, | ||||||
|  |   "interrogate_clip_min_length": 24, | ||||||
|  |   "interrogate_clip_max_length": 48, | ||||||
|  |   "interrogate_clip_dict_limit": 1500.0 | ||||||
|  | } | ||||||
							
								
								
									
										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 7623a5734740025d79b710f3744bff9276e1467b | ||||||
|  | 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=7623a5734740025d79b710f3744bff9276e1467b | ||||||
|  | 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