Stable Diffusion WebUI Docker

Easily and quickly run stable-diffsion-webui with ControlNet in Docker.

Requirements

Docker
Docker Compose

Setup

$ cp .env.example .env

Edit your environment,

$ cat .env

TZ=Etc/UTC                   # timezone
PORT=7861                    # port mapping
VOLUME=data                  # docker volume storage mount point, eg. ./data to /data
DOCKERFILE=Dockerfile        # Specify Dockerfile

If you modified the value of VOLUME, for example from VOLUME=data to VOLUME=volume, please follow these steps as well,

$ cat .env

...
VOLUME=volume   # docker volume storage mount point, eg. ./data to /data

$ mv data volume

Pre-download models and etc.

Here are some models for reference,

$ cat data/enhancing.sh

#!/usr/bin/env bash

declare -A arr

# models
arr["https://huggingface.co/prompthero/openjourney/resolve/main/mdjrny-v4.safetensors"]="models/Stable-diffusion"
arr+=(["https://huggingface.co/prompthero/openjourney-v2/resolve/main/openjourney-v2.ckpt"]="models/Stable-diffusion")
arr+=(["https://huggingface.co/ckpt/anything-v4.5-vae-swapped/resolve/main/anything-v4.5-vae-swapped.safetensors"]="models/Stable-diffusion")
arr+=(["https://huggingface.co/dreamlike-art/dreamlike-photoreal-2.0/resolve/main/dreamlike-photoreal-2.0.safetensors"]="models/Stable-diffusion")
arr+=(["https://huggingface.co/nuigurumi/basil_mix/resolve/main/Basil_mix_fixed.safetensors"]="models/Stable-diffusion")
arr+=(["https://huggingface.co/swl-models/chilloutmix-ni/resolve/main/chilloutmix-Ni-ema-fp32.safetensors"]="models/Stable-diffusion")
# vae
arr+=(["https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors"]="models/VAE")
# embeddings
arr+=(["https://huggingface.co/datasets/gsdf/EasyNegative/resolve/main/EasyNegative.safetensors"]="embeddings")

...

Downloading ...

$ cd data ; bash enhancing.sh

These models, and others. will be linked in Docker once the container is successfully running.

Installation

$ docker compose --profile cpu up -d     # for CPU

$ docker compose --profile gpu up -d     # for GPU

Please wait while the process completes.

If you want to track the progress,

$ docker compose --profile cpu logs -f   # for CPU

$ docker compose --profile gpu logs -f   # for GPU

Usage

Open your browser and navigate to http://127.0.0.1:7861. The default port specified in the .env file is 7861.
The saved images will be located in the data/outputs directory.
Examples of prompts can be found in data/prompts.

Restart, Stop or Deinstallation

$ docker compose --profile cpu restart   # for restart

$ docker compose --profile cpu stop      # for stop

$ docker compose --profile cpu down      # for deinstallation

FAQ

Q: Error message : "RuntimeError: mat1 and mat2 shapes cannot be multiplied (77x1024 and 768x320)"

When encountering the aforementioned error message, please select "v1-5-pruned-emaonly.safetensors" for the "Stable Diffusion checkpoint" to indicate that this operation is currently only supported by Stable Diffusion 1.5.

Q: ERROR: Could not find a version that satisfies the requirement torch==1.13.1+cu117

If you come across the issue outlined in stable-diffusion-webui, a possible solution is to utilize Python 3.10, as described below:

Edit your environment, change DOCKERFILE=Dockerfile to DOCKERFILE=Dockerfile.py310.

$ cat .env

TZ=Etc/UTC                   # timezone
PORT=7861                    # port mapping
VOLUME=data                  # docker volume storage mount point, eg. ./data to /data
DOCKERFILE=Dockerfile.py310  # Specify Dockerfile

The next operation is as usual.

yftzeng / stable-diffusion-webui-docker