mirror of
https://github.com/AbdBarho/stable-diffusion-webui-docker.git
synced 2024-09-11 10:35:57 +03:00
Add AUTOMATIC1111 and lstein WebUIs (#32)
* Lstein * Add AUTOMATIC1111 and lstein UIs * Update Workflow
This commit is contained in:
parent
082876aab3
commit
ccc7306f48
18
.github/workflows/docker.yml
vendored
18
.github/workflows/docker.yml
vendored
@ -2,11 +2,23 @@ name: Build Image
|
||||
|
||||
on: [push]
|
||||
|
||||
# TODO: how to cache intermediate images?
|
||||
jobs:
|
||||
build_docker:
|
||||
build_hlky:
|
||||
runs-on: ubuntu-latest
|
||||
name: Make sure the image builds
|
||||
name: Build hlky
|
||||
steps:
|
||||
- uses: actions/checkout@v3
|
||||
# TODO: how to cache intermediate images?
|
||||
- run: docker compose build --progress plain
|
||||
build_AUTOMATIC1111:
|
||||
runs-on: ubuntu-latest
|
||||
name: Build AUTOMATIC1111
|
||||
steps:
|
||||
- uses: actions/checkout@v3
|
||||
- run: cd AUTOMATIC1111 && docker compose build --progress plain
|
||||
build_lstein:
|
||||
runs-on: ubuntu-latest
|
||||
name: Build lstein
|
||||
steps:
|
||||
- uses: actions/checkout@v3
|
||||
- run: cd lstein && docker compose build --progress plain
|
||||
|
44
AUTOMATIC1111/Dockerfile
Normal file
44
AUTOMATIC1111/Dockerfile
Normal file
@ -0,0 +1,44 @@
|
||||
# syntax=docker/dockerfile:1
|
||||
|
||||
FROM alpine/git:2.36.2 as download
|
||||
RUN <<EOF
|
||||
# who knows
|
||||
git config --global http.postBuffer 1048576000
|
||||
git clone https://github.com/CompVis/stable-diffusion.git repositories/stable-diffusion
|
||||
git clone https://github.com/CompVis/taming-transformers.git repositories/taming-transformers
|
||||
rm -rf repositories/taming-transformers/data repositories/taming-transformers/assets
|
||||
EOF
|
||||
|
||||
FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime
|
||||
|
||||
SHELL ["/bin/bash", "-ceuxo", "pipefail"]
|
||||
|
||||
ENV DEBIAN_FRONTEND=noninteractive
|
||||
RUN apt-get update && apt-get install git -y && apt-get clean
|
||||
|
||||
RUN <<EOF
|
||||
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
|
||||
cd stable-diffusion-webui
|
||||
git reset --hard 064965c4660f57f24e2d51a9854defaeabf8c0cf
|
||||
pip install -U --prefer-binary --no-cache-dir -r requirements.txt
|
||||
EOF
|
||||
|
||||
RUN <<EOF
|
||||
pip install --prefer-binary -U --no-cache-dir diffusers numpy invisible-watermark git+https://github.com/crowsonkb/k-diffusion.git \
|
||||
git+https://github.com/TencentARC/GFPGAN.git markupsafe==2.0.1 opencv-python-headless
|
||||
EOF
|
||||
|
||||
|
||||
ENV ROOT=/workspace/stable-diffusion-webui \
|
||||
WORKDIR=/workspace/stable-diffusion-webui/repositories/stable-diffusion \
|
||||
TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS=""
|
||||
|
||||
COPY --from=download /git/ ${ROOT}
|
||||
|
||||
|
||||
COPY . /docker
|
||||
|
||||
WORKDIR ${WORKDIR}
|
||||
EXPOSE 7860
|
||||
# run, -u to not buffer stdout / stderr
|
||||
CMD /docker/mount.sh && python3 -u ../../webui.py --listen ${CLI_ARGS}
|
12
AUTOMATIC1111/README.md
Normal file
12
AUTOMATIC1111/README.md
Normal file
@ -0,0 +1,12 @@
|
||||
# WebUI for AUTOMATIC1111
|
||||
|
||||
The WebUI of [AUTOMATIC1111/stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui) as docker container!
|
||||
|
||||
## Setup
|
||||
|
||||
Clone this repo, download the `model.ckpt` and `GFPGANv1.3.pth` and put into the `models` folder as mentioned in [the main README](../README.md), then run
|
||||
|
||||
```
|
||||
cd AUTOMATIC1111
|
||||
docker compose up --build
|
||||
```
|
3
AUTOMATIC1111/config.json
Normal file
3
AUTOMATIC1111/config.json
Normal file
@ -0,0 +1,3 @@
|
||||
{
|
||||
"outdir_samples": "/output"
|
||||
}
|
20
AUTOMATIC1111/docker-compose.yml
Normal file
20
AUTOMATIC1111/docker-compose.yml
Normal file
@ -0,0 +1,20 @@
|
||||
version: '3.9'
|
||||
|
||||
services:
|
||||
model:
|
||||
build: .
|
||||
ports:
|
||||
- "7860:7860"
|
||||
volumes:
|
||||
- ../cache:/cache
|
||||
- ../output:/output
|
||||
- ../models:/models
|
||||
environment:
|
||||
- CLI_ARGS=--medvram --opt-split-attention
|
||||
deploy:
|
||||
resources:
|
||||
reservations:
|
||||
devices:
|
||||
- driver: nvidia
|
||||
device_ids: ['0']
|
||||
capabilities: [gpu]
|
28
AUTOMATIC1111/mount.sh
Normal file
28
AUTOMATIC1111/mount.sh
Normal file
@ -0,0 +1,28 @@
|
||||
#!/bin/bash
|
||||
|
||||
declare -A MODELS
|
||||
|
||||
MODELS["${WORKDIR}/models/ldm/stable-diffusion-v1/model.ckpt"]=model.ckpt
|
||||
MODELS["${ROOT}/GFPGANv1.3.pth"]=GFPGANv1.3.pth
|
||||
|
||||
for path in "${!MODELS[@]}"; do
|
||||
name=${MODELS[$path]}
|
||||
base=$(dirname "${path}")
|
||||
from_path="/models/${name}"
|
||||
if test -f "${from_path}"; then
|
||||
mkdir -p "${base}" && ln -sf "${from_path}" "${path}" && echo "Mounted ${name}"
|
||||
else
|
||||
echo "Skipping ${name}"
|
||||
fi
|
||||
done
|
||||
|
||||
# force realesrgan cache
|
||||
rm -rf /opt/conda/lib/python3.7/site-packages/realesrgan/weights
|
||||
ln -s -T /models /opt/conda/lib/python3.7/site-packages/realesrgan/weights
|
||||
|
||||
# force facexlib cache
|
||||
mkdir -p /cache/weights/ ${WORKDIR}/gfpgan/
|
||||
ln -sf /cache/weights/ ${WORKDIR}/gfpgan/
|
||||
|
||||
# mount config
|
||||
ln -sf /docker/config.json ${WORKDIR}/config.json
|
@ -2,7 +2,12 @@
|
||||
|
||||
Run Stable Diffusion on your machine with a nice UI without any hassle!
|
||||
|
||||
This repository provides the [WebUI](https://github.com/hlky/stable-diffusion-webui) as a docker image for easy setup and deployment. Please note that the WebUI is experimental and evolving quickly, so expect some bugs.
|
||||
This repository provides the [WebUI](https://github.com/hlky/stable-diffusion-webui) as a docker image for easy setup and deployment.
|
||||
|
||||
Now with experimental support for 2 other forks:
|
||||
|
||||
- [AUTOMATIC1111](./AUTOMATIC1111/) (Stable, very few bugs!)
|
||||
- [lstein](./lstein/)
|
||||
|
||||
## Features
|
||||
|
||||
@ -13,7 +18,6 @@ This repository provides the [WebUI](https://github.com/hlky/stable-diffusion-we
|
||||
- Latent Diffusion Super Resolution
|
||||
- GoBig
|
||||
- GoLatent
|
||||
- [Textual Inversion](https://github.com/hlky/sd-enable-textual-inversion)
|
||||
- many more!
|
||||
|
||||
## Setup
|
||||
@ -79,6 +83,7 @@ Special thanks to everyone behind these awesome projects, without them, none of
|
||||
|
||||
- [hlky/stable-diffusion-webui](https://github.com/hlky/stable-diffusion-webui)
|
||||
- [AUTOMATIC1111/stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
|
||||
- [lstein/stable-diffusion](https://github.com/lstein/stable-diffusion)
|
||||
- [CompVis/stable-diffusion](https://github.com/CompVis/stable-diffusion)
|
||||
- [hlky/sd-enable-textual-inversion](https://github.com/hlky/sd-enable-textual-inversion)
|
||||
- [devilismyfriend/latent-diffusion](https://github.com/devilismyfriend/latent-diffusion)
|
||||
|
@ -3,7 +3,7 @@ version: '3.9'
|
||||
services:
|
||||
model:
|
||||
build:
|
||||
context: ./build/
|
||||
context: ./hlky/
|
||||
args:
|
||||
# You can choose any commit sha from https://github.com/hlky/stable-diffusion/commits/main
|
||||
# USE AT YOUR OWN RISK! otherwise just leave it empty.
|
||||
|
0
build/mount.sh → hlky/mount.sh
Executable file → Normal file
0
build/mount.sh → hlky/mount.sh
Executable file → Normal file
29
lstein/Dockerfile
Normal file
29
lstein/Dockerfile
Normal file
@ -0,0 +1,29 @@
|
||||
# syntax=docker/dockerfile:1
|
||||
|
||||
FROM continuumio/miniconda3:4.12.0
|
||||
|
||||
SHELL ["/bin/bash", "-ceuxo", "pipefail"]
|
||||
|
||||
RUN conda install python=3.8.5 && conda clean -a -y
|
||||
RUN conda install pytorch==1.11.0 torchvision==0.12.0 cudatoolkit=11.3 -c pytorch && conda clean -a -y
|
||||
|
||||
RUN apt-get update && apt install fonts-dejavu-core rsync -y && apt-get clean
|
||||
|
||||
|
||||
RUN <<EOF
|
||||
git clone https://github.com/lstein/stable-diffusion.git
|
||||
cd stable-diffusion
|
||||
git reset --hard 751283a2de81bee4bb571fbabe4adb19f1d85b97
|
||||
conda env update --file environment.yaml -n base
|
||||
conda clean -a -y
|
||||
EOF
|
||||
|
||||
ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS=""
|
||||
|
||||
WORKDIR /stable-diffusion
|
||||
|
||||
EXPOSE 7860
|
||||
# run, -u to not buffer stdout / stderr
|
||||
CMD mkdir -p /stable-diffusion/models/ldm/stable-diffusion-v1/ && \
|
||||
ln -sf /models/model.ckpt /stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt && \
|
||||
python3 -u scripts/dream.py --outdir /output --web --host 0.0.0.0 --port 7860 ${CLI_ARGS}
|
14
lstein/README.md
Normal file
14
lstein/README.md
Normal file
@ -0,0 +1,14 @@
|
||||
# WebUI for lstein
|
||||
|
||||
The WebUI of [lstein/stable-diffusion](https://github.com/lstein/stable-diffusion) as docker container!
|
||||
|
||||
Although it is a simple UI, the project has a lot of potential.
|
||||
|
||||
## Setup
|
||||
|
||||
Clone this repo, download the `model.ckpt` and put into the `models` folder as mentioned in [the main README](../README.md), then run
|
||||
|
||||
```
|
||||
cd lstein
|
||||
docker compose up --build
|
||||
```
|
21
lstein/docker-compose.yml
Normal file
21
lstein/docker-compose.yml
Normal file
@ -0,0 +1,21 @@
|
||||
version: '3.9'
|
||||
|
||||
services:
|
||||
model:
|
||||
build: .
|
||||
restart: on-failure
|
||||
ports:
|
||||
- "7860:7860"
|
||||
volumes:
|
||||
- ../cache:/cache
|
||||
- ../output:/output
|
||||
- ../models:/models
|
||||
environment:
|
||||
- CLI_ARGS=
|
||||
deploy:
|
||||
resources:
|
||||
reservations:
|
||||
devices:
|
||||
- driver: nvidia
|
||||
device_ids: ['0']
|
||||
capabilities: [gpu]
|
1
models/.gitignore
vendored
1
models/.gitignore
vendored
@ -1,5 +1,6 @@
|
||||
/model.ckpt
|
||||
/GFPGANv1.3.pth
|
||||
/RealESRGAN_x2plus.pth
|
||||
/RealESRGAN_x4plus.pth
|
||||
/RealESRGAN_x4plus_anime_6B.pth
|
||||
/LDSR.ckpt
|
||||
|
Loading…
Reference in New Issue
Block a user