• Stars
    star
    128
  • Rank 281,044 (Top 6 %)
  • Language
    Dockerfile
  • License
    MIT License
  • Created almost 2 years ago
  • Updated over 1 year ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

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

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)"

RuntimeError: mat1 and mat2 shapes cannot be multiplied

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.