From b22479e177e4fb1a086ae81b26a0d8d683a63503 Mon Sep 17 00:00:00 2001
From: AbdBarho <ka70911@gmail.com>
Date: Fri, 7 Oct 2022 09:18:31 +0200
Subject: [PATCH] lstein

---
 services/lstein/Dockerfile | 22 ++++++++++-----------
 services/lstein/info.py    |  7 +++++--
 services/lstein/mount.sh   | 39 +++++++++++++++++++-------------------
 3 files changed, 34 insertions(+), 34 deletions(-)

diff --git a/services/lstein/Dockerfile b/services/lstein/Dockerfile
index 5944b13..7b8d65b 100644
--- a/services/lstein/Dockerfile
+++ b/services/lstein/Dockerfile
@@ -12,21 +12,20 @@ RUN conda install pytorch==1.11.0 torchvision==0.12.0 cudatoolkit=11.3 -c pytorc
 
 RUN apt-get update && apt install fonts-dejavu-core rsync gcc -y && apt-get clean
 
+ENV PIP_EXISTS_ACTION=w
 
 RUN <<EOF
 git clone https://github.com/invoke-ai/InvokeAI.git stable-diffusion
 cd stable-diffusion
-git reset --hard a1739a73b48bfe98b6abcb67f5a0197a9ad270e0
+git reset --hard 8a8be92eac17e0ef699528157596b2336bdee532
 sed -i -- 's/python=3.8.5/python=3.9/g' environment.yaml
+git config --global http.postBuffer 1048576000
 conda env update --file environment.yaml -n base
 conda clean -a -y
 EOF
 
 
-ARG BRANCH=development SHA=8c9f2ae705cf723d4a8a73c416e8d8bf2d746977
-# this breaks on generation:
-# there is a new UI anyway, but it is not by any means ready.
-# ARG BRANCH=development SHA=bdbc76fcd4bd3362312dc91b087d9af66de423b1
+ARG BRANCH=development SHA=4f247a3672474bd9c46060bab6087dbf9e2531f3
 RUN <<EOF
 cd stable-diffusion
 git fetch
@@ -41,19 +40,18 @@ RUN pip uninstall opencv-python -y && pip install --prefer-binary --force-reinst
 
 COPY . /docker/
 RUN <<EOF
-python3 /docker/info.py /stable-diffusion/static/dream_web/index.html
+python3 /docker/info.py /stable-diffusion/frontend/dist/index.html
 chmod +x /docker/mount.sh
-sed -i -- 's/outputs\//\/output/g' /stable-diffusion/backend/server.py
 EOF
 
 
-ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch PRELOAD=false CLI_ARGS=""
+ENV PRELOAD=false CLI_ARGS=""
 WORKDIR /stable-diffusion
 EXPOSE 7860
 
 CMD /docker/mount.sh && \
   # python3 -u backend/server.py --host 0.0.0.0 --port 7860 --cors http://localhost:7860
-  # python3 -u scripts/dream.py --outdir /output --web --host 0.0.0.0 --port 7860 ${CLI_ARGS}
-  echo The lstein webUI is currently deactivated due to implementation limitations: \
-  https://github.com/invoke-ai/InvokeAI/blob/8c9f2ae705cf723d4a8a73c416e8d8bf2d746977/backend/modules/create_cmd_parser.py#L26 \
-  Once the path the output is fixed, the UI will be activated again
+  python3 -u scripts/dream.py --outdir /output --web --host 0.0.0.0 --port 7860 ${CLI_ARGS}
+# echo The lstein webUI is currently deactivated due to implementation limitations: \
+# https://github.com/invoke-ai/InvokeAI/blob/8c9f2ae705cf723d4a8a73c416e8d8bf2d746977/backend/modules/create_cmd_parser.py#L26 \
+# Once the path the output is fixed, the UI will be activated again
diff --git a/services/lstein/info.py b/services/lstein/info.py
index 1d8f476..4b8ece8 100644
--- a/services/lstein/info.py
+++ b/services/lstein/info.py
@@ -4,7 +4,10 @@ from pathlib import Path
 file = Path(sys.argv[1])
 file.write_text(
   file.read_text()\
-  .replace('GitHub site</a>', """
-  GitHub site</a>, Deployed with <a href="https://github.com/AbdBarho/stable-diffusion-webui-docker/">stable-diffusion-webui-docker</a>
+  .replace('  <div id="root"></div>', """
+    <div id="root"></div>
+    <div>
+      Deployed with <a href="https://github.com/AbdBarho/stable-diffusion-webui-docker/">stable-diffusion-webui-docker</a>
+    </div>
 """, 1)
 )
diff --git a/services/lstein/mount.sh b/services/lstein/mount.sh
index 0b31bdf..808c468 100755
--- a/services/lstein/mount.sh
+++ b/services/lstein/mount.sh
@@ -1,28 +1,27 @@
 #!/bin/bash
 
-set -eu
+set -Eeuo pipefail
 
-ROOT=/stable-diffusion
+declare -A MOUNTS
 
-mkdir -p "${ROOT}/models/ldm/stable-diffusion-v1/"
-ln -sf /cache/models/model.ckpt "${ROOT}/models/ldm/stable-diffusion-v1/model.ckpt"
+# cache
+MOUNTS["/root/.cache"]=/data/.cache
+# ui specific
+MOUNTS["${PWD}/models/ldm/stable-diffusion-v1/model.ckpt"]=/data/StableDiffusion/model.ckpt
+MOUNTS["${PWD}/src/gfpgan/experiments/pretrained_models/GFPGANv1.4.pth"]=/data/GFPGAN/GFPGANv1.4.pth
+MOUNTS["${PWD}/ldm/dream/restoration/codeformer/weights"]=/data/CodeFormer
+# hacks
+MOUNTS["/opt/conda/lib/python3.9/site-packages/facexlib/weights"]=/data/.cache
+MOUNTS["/opt/conda/lib/python3.9/site-packages/realesrgan/weights"]=/data/RealESRGAN
 
-base="${ROOT}/src/gfpgan/experiments/pretrained_models/"
-mkdir -p "${base}"
-# TODO: "real" GFPGANv1.4.pth
-ln -sf /cache/models/GFPGANv1.3.pth "${base}/GFPGANv1.4.pth"
-echo "Mounted GFPGANv1.3.pth"
-
-# facexlib
-FACEX_WEIGHTS=/opt/conda/lib/python3.9/site-packages/facexlib/weights
-
-rm -rf "${FACEX_WEIGHTS}"
-mkdir -p /cache/weights
-ln -sf -T /cache/weights "${FACEX_WEIGHTS}"
-
-REALESRGAN_WEIGHTS=/opt/conda/lib/python3.9/site-packages/realesrgan/weights
-rm -rf "${REALESRGAN_WEIGHTS}"
-ln -sf -T /cache/weights "${REALESRGAN_WEIGHTS}"
+for to_path in "${!MOUNTS[@]}"; do
+  set -Eeuo pipefail
+  from_path="${MOUNTS[${to_path}]}"
+  rm -rf "${to_path}"
+  mkdir -p "$(dirname "${to_path}")"
+  ln -sT "${from_path}" "${to_path}"
+  echo Mounted $(basename "${from_path}")
+done
 
 if "${PRELOAD}" == "true"; then
   python3 -u scripts/preload_models.py