https://github.com/ashleykleynhans/stable-diffusion-docker
Docker image for Stable Diffusion WebUI with ControlNet, After Detailer, Dreambooth, Deforum and ReActor extensions, as well as Kohya_ss and ComfyUI
https://github.com/ashleykleynhans/stable-diffusion-docker
adetailer comfyui controlnet deforum deforum-stable-diffusion docker dreambooth face-swap kohya-webui roop runpod sdxl sdxl-docker stable-diffusion stable-diffusion-webui
Last synced: about 2 months ago
JSON representation
Docker image for Stable Diffusion WebUI with ControlNet, After Detailer, Dreambooth, Deforum and ReActor extensions, as well as Kohya_ss and ComfyUI
- Host: GitHub
- URL: https://github.com/ashleykleynhans/stable-diffusion-docker
- Owner: ashleykleynhans
- License: gpl-3.0
- Created: 2023-05-29T14:04:27.000Z (almost 2 years ago)
- Default Branch: main
- Last Pushed: 2024-08-05T15:20:13.000Z (9 months ago)
- Last Synced: 2024-08-06T16:55:33.117Z (9 months ago)
- Topics: adetailer, comfyui, controlnet, deforum, deforum-stable-diffusion, docker, dreambooth, face-swap, kohya-webui, roop, runpod, sdxl, sdxl-docker, stable-diffusion, stable-diffusion-webui
- Language: Shell
- Homepage:
- Size: 682 KB
- Stars: 336
- Watchers: 12
- Forks: 93
- Open Issues: 4
-
Metadata Files:
- Readme: README.md
- Funding: .github/FUNDING.yml
- License: LICENSE
Awesome Lists containing this project
- awesome-ccamel - ashleykleynhans/stable-diffusion-docker - Docker image for Stable Diffusion WebUI with ControlNet, After Detailer, Dreambooth, Deforum and ReActor extensions, as well as Kohya_ss and ComfyUI (Shell)
README
# Docker image for A1111 Stable Diffusion Web UI, Kohya_ss, ComfyUI and InvokeAI
[](https://github.com/ashleykleynhans/stable-diffusion-docker)
[](https://hub.docker.com/repository/docker/ashleykza/stable-diffusion-webui)
[](https://runpod.io/gsc?template=ya6013lj5a&ref=2xxro4sy)
Now with SDXL support.
## Installs
* Ubuntu 22.04 LTS
* CUDA 12.1
* Python 3.10.12
* Torch 2.4.0
* xformers 0.0.27.post2
* [Jupyter Lab](https://github.com/jupyterlab/jupyterlab)
* [code-server](https://github.com/coder/code-server)
* [Automatic1111 Stable Diffusion Web UI](
https://github.com/AUTOMATIC1111/stable-diffusion-webui) 1.10.0
* [ControlNet extension](
https://github.com/Mikubill/sd-webui-controlnet) v1.1.455
* [After Detailer extension](
https://github.com/Bing-su/adetailer) v24.6.0
* [Locon extension](
https://github.com/ashleykleynhans/a1111-sd-webui-locon)
* [ReActor extension](https://github.com/Gourieff/sd-webui-reactor)
* [Deforum extension](https://github.com/deforum-art/sd-webui-deforum)
* [Inpaint Anything extension](https://github.com/Uminosachi/sd-webui-inpaint-anything)
* [Infinite Image Browsing extension](https://github.com/zanllp/sd-webui-infinite-image-browsing)
* [CivitAI extension](https://github.com/civitai/sd_civitai_extension)
* [CivitAI Browser+ extension](https://github.com/BlafKing/sd-civitai-browser-plus)
* [TensorRT extension](https://github.com/NVIDIA/Stable-Diffusion-WebUI-TensorRT)
* [Stable Diffusion Dynamic Thresholding (CFG Scale Fix) extension](https://github.com/mcmonkeyprojects/sd-dynamic-thresholding)
* [Kohya_ss](https://github.com/bmaltais/kohya_ss) v24.1.6
* [ComfyUI](https://github.com/comfyanonymous/ComfyUI)
* [ComfyUI Manager](https://github.com/ltdrdata/ComfyUI-Manager)
* [InvokeAI](https://github.com/invoke-ai/InvokeAI) v4.2.7post1
* [sd_xl_base_1.0.safetensors](
https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors)
* [sd_xl_refiner_1.0.safetensors](
https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0/resolve/main/sd_xl_refiner_1.0.safetensors)
* [sdxl_vae.safetensors](
https://huggingface.co/madebyollin/sdxl-vae-fp16-fix/resolve/main/sdxl_vae.safetensors)
* [inswapper_128.onnx](
https://github.com/facefusion/facefusion-assets/releases/download/models/inswapper_128.onnx)
* [runpodctl](https://github.com/runpod/runpodctl)
* [OhMyRunPod](https://github.com/kodxana/OhMyRunPod)
* [RunPod File Uploader](https://github.com/kodxana/RunPod-FilleUploader)
* [croc](https://github.com/schollz/croc)
* [rclone](https://rclone.org/)
* [Application Manager](https://github.com/ashleykleynhans/app-manager)
* [CivitAI Downloader](https://github.com/ashleykleynhans/civitai-downloader)
## Available on RunPod
This image is designed to work on [RunPod](https://runpod.io?ref=2xxro4sy).
You can use my custom [RunPod template](
https://runpod.io/gsc?template=ya6013lj5a&ref=2xxro4sy)
to launch it on RunPod.
## Building the Docker image
> [!NOTE]
> You will need to edit the `docker-bake.hcl` file and update `REGISTRY_USER`,
> and `RELEASE`. You can obviously edit the other values too, but these
> are the most important ones.
> [!IMPORTANT]
> In order to cache the models, you will need at least 32GB of CPU/system
> memory (not VRAM) due to the large size of the models. If you have less
> than 32GB of system memory, you can comment out or remove the code in the
> `Dockerfile` that caches the models.
```bash
# Clone the repo
git clone https://github.com/ashleykleynhans/stable-diffusion-docker.git
# Download the models
cd stable-diffusion-docker
wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned.safetensors
wget https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors
wget https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors
wget https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0/resolve/main/sd_xl_refiner_1.0.safetensors
wget https://huggingface.co/madebyollin/sdxl-vae-fp16-fix/resolve/main/sdxl_vae.safetensors
# Log in to Docker Hub
docker login
# Build the image, tag the image, and push the image to Docker Hub
docker buildx bake -f docker-bake.hcl --push
# Same as above but customize registry/user/release:
REGISTRY=ghcr.io REGISTRY_USER=myuser RELEASE=my-release docker buildx \
bake -f docker-bake.hcl --push
```
## Running Locally
### Install Nvidia CUDA Driver
- [Linux](https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html)
- [Windows](https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html)
### Start the Docker container
```bash
docker run -d \
--gpus all \
-v /workspace \
-p 2999:2999 \
-p 3000:3001 \
-p 3010:3011 \
-p 3020:3021 \
-p 6006:6066 \
-p 7777:7777 \
-p 8000:8000 \
-p 8888:8888 \
-p 9090:9090 \
-e VENV_PATH=/workspace/venvs/a1111 \
-e JUPYTER_PASSWORD=Jup1t3R! \
-e ENABLE_TENSORBOARD=1 \
ashleykza/stable-diffusion-webui:latest
```
You can obviously substitute the image name and tag with your own.
### Ports
| Connect Port | Internal Port | Description |
|--------------|---------------|-------------------------------|
| 3000 | 3001 | A1111 Stable Diffusion Web UI |
| 3010 | 3011 | Kohya_ss |
| 3020 | 3021 | ComfyUI |
| 9090 | 9090 | InvokeAI |
| 6006 | 6066 | Tensorboard |
| 7777 | 7777 | Code Server |
| 8000 | 8000 | Application Manager |
| 8888 | 8888 | Jupyter Lab |
| 2999 | 2999 | RunPod File Uploader |
### Environment Variables
| Variable | Description | Default |
|----------------------|--------------------------------------------------|------------------------|
| VENV_PATH | Set the path for the Python venv for the app | /workspace/venvs/a1111 |
| JUPYTER_LAB_PASSWORD | Set a password for Jupyter lab | not set - no password |
| DISABLE_AUTOLAUNCH | Disable Web UIs from launching automatically | (not set) |
| DISABLE_SYNC | Disable syncing if using a RunPod network volume | (not set) |
| ENABLE_TENSORBOARD | Enables Tensorboard on port 6006 | enabled |
## Logs
Stable Diffusion Web UI, Kohya SS, ComfyUI, and InvokeAI each
create log files, and you can tail the log files instead of
killing the services to view the logs
| Application | Log file |
|-------------------------|------------------------------|
| Stable Diffusion Web UI | /workspace/logs/webui.log |
| Kohya SS | /workspace/logs/kohya_ss.log |
| ComfyUI | /workspace/logs/comfyui.log |
| InvokeAI | /workspace/logs/invokeai.log |
For example:
```bash
tail -f /workspace/logs/webui.log
```
## Community and Contributing
Pull requests and issues on [GitHub](https://github.com/ashleykleynhans/stable-diffusion-docker)
are welcome. Bug fixes and new features are encouraged.
## Appreciate my work?
