From ccc7306f48106e385486de45185eee1f68e7d151 Mon Sep 17 00:00:00 2001
From: AbdBarho <ka70911@gmail.com>
Date: Mon, 5 Sep 2022 19:51:22 +0200
Subject: [PATCH] Add AUTOMATIC1111 and lstein WebUIs (#32)

* Lstein

* Add AUTOMATIC1111 and lstein UIs

* Update Workflow
---
 .github/workflows/docker.yml     | 18 ++++++++++---
 AUTOMATIC1111/Dockerfile         | 44 ++++++++++++++++++++++++++++++++
 AUTOMATIC1111/README.md          | 12 +++++++++
 AUTOMATIC1111/config.json        |  3 +++
 AUTOMATIC1111/docker-compose.yml | 20 +++++++++++++++
 AUTOMATIC1111/mount.sh           | 28 ++++++++++++++++++++
 README.md                        |  9 +++++--
 docker-compose.yml               |  2 +-
 {build => hlky}/Dockerfile       |  0
 {build => hlky}/info.py          |  0
 {build => hlky}/mount.sh         |  0
 lstein/Dockerfile                | 29 +++++++++++++++++++++
 lstein/README.md                 | 14 ++++++++++
 lstein/docker-compose.yml        | 21 +++++++++++++++
 models/.gitignore                |  1 +
 15 files changed, 195 insertions(+), 6 deletions(-)
 create mode 100644 AUTOMATIC1111/Dockerfile
 create mode 100644 AUTOMATIC1111/README.md
 create mode 100644 AUTOMATIC1111/config.json
 create mode 100644 AUTOMATIC1111/docker-compose.yml
 create mode 100644 AUTOMATIC1111/mount.sh
 rename {build => hlky}/Dockerfile (100%)
 rename {build => hlky}/info.py (100%)
 rename {build => hlky}/mount.sh (100%)
 mode change 100755 => 100644
 create mode 100644 lstein/Dockerfile
 create mode 100644 lstein/README.md
 create mode 100644 lstein/docker-compose.yml

diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index 78e7e01..3d6a0ce 100644
--- a/.github/workflows/docker.yml
+++ b/.github/workflows/docker.yml
@@ -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
diff --git a/AUTOMATIC1111/Dockerfile b/AUTOMATIC1111/Dockerfile
new file mode 100644
index 0000000..d7027f4
--- /dev/null
+++ b/AUTOMATIC1111/Dockerfile
@@ -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}
diff --git a/AUTOMATIC1111/README.md b/AUTOMATIC1111/README.md
new file mode 100644
index 0000000..151bf75
--- /dev/null
+++ b/AUTOMATIC1111/README.md
@@ -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
+```
diff --git a/AUTOMATIC1111/config.json b/AUTOMATIC1111/config.json
new file mode 100644
index 0000000..ef09c9a
--- /dev/null
+++ b/AUTOMATIC1111/config.json
@@ -0,0 +1,3 @@
+{
+  "outdir_samples": "/output"
+}
diff --git a/AUTOMATIC1111/docker-compose.yml b/AUTOMATIC1111/docker-compose.yml
new file mode 100644
index 0000000..0964f90
--- /dev/null
+++ b/AUTOMATIC1111/docker-compose.yml
@@ -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]
diff --git a/AUTOMATIC1111/mount.sh b/AUTOMATIC1111/mount.sh
new file mode 100644
index 0000000..71e6d80
--- /dev/null
+++ b/AUTOMATIC1111/mount.sh
@@ -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
diff --git a/README.md b/README.md
index f0ef149..926740d 100644
--- a/README.md
+++ b/README.md
@@ -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)
diff --git a/docker-compose.yml b/docker-compose.yml
index 2563f30..214aa66 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -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.
diff --git a/build/Dockerfile b/hlky/Dockerfile
similarity index 100%
rename from build/Dockerfile
rename to hlky/Dockerfile
diff --git a/build/info.py b/hlky/info.py
similarity index 100%
rename from build/info.py
rename to hlky/info.py
diff --git a/build/mount.sh b/hlky/mount.sh
old mode 100755
new mode 100644
similarity index 100%
rename from build/mount.sh
rename to hlky/mount.sh
diff --git a/lstein/Dockerfile b/lstein/Dockerfile
new file mode 100644
index 0000000..dd5d7a8
--- /dev/null
+++ b/lstein/Dockerfile
@@ -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}
diff --git a/lstein/README.md b/lstein/README.md
new file mode 100644
index 0000000..21d4b5b
--- /dev/null
+++ b/lstein/README.md
@@ -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
+```
diff --git a/lstein/docker-compose.yml b/lstein/docker-compose.yml
new file mode 100644
index 0000000..4f3c918
--- /dev/null
+++ b/lstein/docker-compose.yml
@@ -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]
diff --git a/models/.gitignore b/models/.gitignore
index 94fbb12..8d8e32c 100644
--- a/models/.gitignore
+++ b/models/.gitignore
@@ -1,5 +1,6 @@
 /model.ckpt
 /GFPGANv1.3.pth
+/RealESRGAN_x2plus.pth
 /RealESRGAN_x4plus.pth
 /RealESRGAN_x4plus_anime_6B.pth
 /LDSR.ckpt