Stable Diffusion WebUI Docker

Run Stable Diffusion on your machine with a nice UI without any hassle!

This repository provides the WebUI as a docker image for easy setup and deployment. Please note that the WebUI is experimental and evolving quickly, so expect some bugs.

Features

• Interactive UI with many features, and more on the way!
• Support for 6GB GPU cards.
• GFPGAN for face reconstruction, RealESRGAN for super-sampling.
• Experimental:
  • Latent Diffusion Super Resolution
  • GoBig
  • GoLatent
  • Textual Inversion
• many more!

Setup

Make sure you have an up to date version of docker installed. Download this repo and run:

docker compose build

you can let it build in the background while you download the different models

• Stable Diffusion v1.4 (4GB), rename to model.ckpt
• (Optional) GFPGANv1.3.pth (333MB).
• (Optional) RealESRGAN_x4plus.pth (64MB) and RealESRGAN_x4plus_anime_6B.pth (18MB).
• (Optional) LDSR (2GB) and its configuration, rename to LDSR.ckpt and LDSR.yaml respectively.

Put all of the downloaded files in the models folder, it should look something like this:

models/
├── model.ckpt
├── GFPGANv1.3.pth
├── RealESRGAN_x4plus.pth
├── RealESRGAN_x4plus_anime_6B.pth
├── LDSR.ckpt
└── LDSR.yaml

Run

After the build is done, you can run the app with:

docker compose up --build

Will start the app on http://localhost:7860/

Note: the first start will take sometime as some other models will be downloaded, these will be cached in the cache folder, so next runs are faster.

FAQ

You can find fixes to common issues in the wiki page.

Config

in the docker-compose.yml you can change the CLI_ARGS variable, which contains the arguments that will be passed to the WebUI. By default: --extra-models-cpu --optimized-turbo are given, which allow you to use this model on a 6GB GPU. However, some features might not be available in the mode. You can find the full list of arguments here.

You can set the WEBUI_SHA to any SHA from the main repo, this will build the container against that commit. Use at your own risk.

Disclaimer

The authors of this project are not responsible for any content generated using this interface.

This license of this software forbids you from sharing any content that violates any laws, produce any harm to a person, disseminate any personal information that would be meant for harm, spread misinformation and target vulnerable groups. For the full list of restrictions please read the license.

Thanks

Special thanks to everyone behind these awesome projects, without them, none of this would have been possible:

• hlky/stable-diffusion-webui
• AUTOMATIC1111/stable-diffusion-webui
• CompVis/stable-diffusion
• hlky/sd-enable-textual-inversion
• devilismyfriend/latent-diffusion