Stable Diffusion WebUI Docker

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

This repository provides the WebUI as docker for easy setup and deployment. Please note that this repo delivers all cutting-edge unstable changes from the WebUI, 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.
Textual Inversion
many more!

Setup

make sure you have docker installed and up to date. 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).

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

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

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.

Config

in the docker-compose.yml you can change the CLI_ARGS variable contains all of the variables that will be passed to the web ui.

By default: --extra-models-cpu --optimized-turbo are given, which allow you to use this model on a 6GB GPU.

FAQ

Enable Textual Inversion remove --optimize and --optimize-turbo flags and add --no-half, more info here.
The Application Freezes When Trying Batch Sizes Higher Than One
Output is a green imagee

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:

3.5 KiB Raw Blame History