From f1a1641add32f7a47e584d75836bdd9e4fbe0caf Mon Sep 17 00:00:00 2001
From: AbdBarho <ka70911@gmail.com>
Date: Sat, 3 Dec 2022 13:36:18 +0100
Subject: [PATCH] Invoke 2.2.3 (#250)

lstein:
https://github.com/invoke-ai/InvokeAI/commit/5c31feb3a1096d437c94b6e1c3224eb7a7224a85
---
 docker-compose.yml                |  8 ++++----
 services/AUTOMATIC1111/Dockerfile |  2 +-
 services/hlky/Dockerfile          |  2 ++
 services/lstein/Dockerfile        | 30 ++++++++++++++++++++----------
 services/lstein/entrypoint.sh     |  5 ++---
 5 files changed, 29 insertions(+), 18 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 068efdd..78e86a7 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -27,7 +27,7 @@ services:
     <<: *base_service
     profiles: ["auto"]
     build: ./services/AUTOMATIC1111
-    image: sd-auto:18
+    image: sd-auto:19
     environment:
       - CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access
 
@@ -42,7 +42,7 @@ services:
     <<: *base_service
     profiles: ["hlky"]
     build: ./services/hlky/
-    image: sd-hlky:10
+    image: sd-hlky:11
     environment:
       - CLI_ARGS=--optimized-turbo
       - USE_STREAMLIT=0
@@ -51,7 +51,7 @@ services:
     <<: *base_service
     profiles: ["lstein"]
     build: ./services/lstein/
-    image: sd-lstein:8
+    image: sd-lstein:9
     environment:
       - PRELOAD=true
-      - CLI_ARGS=--max_loaded_models=1
+      - CLI_ARGS=
diff --git a/services/AUTOMATIC1111/Dockerfile b/services/AUTOMATIC1111/Dockerfile
index 3dce5c6..e937a52 100644
--- a/services/AUTOMATIC1111/Dockerfile
+++ b/services/AUTOMATIC1111/Dockerfile
@@ -78,7 +78,7 @@ git reset --hard ${SHA}
 pip install -r requirements_versions.txt
 EOF
 
-RUN pip install opencv-python-headless
+RUN pip install opencv-python-headless transformers==4.24.0
 
 COPY . /docker
 
diff --git a/services/hlky/Dockerfile b/services/hlky/Dockerfile
index d53d5d6..c8b3f0d 100644
--- a/services/hlky/Dockerfile
+++ b/services/hlky/Dockerfile
@@ -35,6 +35,8 @@ conda env update --file environment.yaml -n base
 conda clean -a -y
 EOF
 
+RUN pip install transformers==4.24.0
+
 # add info
 COPY . /docker/
 RUN <<EOF
diff --git a/services/lstein/Dockerfile b/services/lstein/Dockerfile
index c4bd696..457c551 100644
--- a/services/lstein/Dockerfile
+++ b/services/lstein/Dockerfile
@@ -1,13 +1,12 @@
 # syntax=docker/dockerfile:1
 
 FROM python:3.10-slim
-
 SHELL ["/bin/bash", "-ceuxo", "pipefail"]
 
 ENV DEBIAN_FRONTEND=noninteractive PIP_EXISTS_ACTION=w  PIP_PREFER_BINARY=1 PIP_NO_CACHE_DIR=1
 
 
-RUN pip install torch==1.13.0 torchvision --extra-index-url https://download.pytorch.org/whl/cu117
+RUN pip install torch==1.12.0+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
 
 RUN apt-get update && apt-get install git -y && apt-get clean
 
@@ -16,23 +15,34 @@ RUN git clone https://github.com/invoke-ai/InvokeAI.git /stable-diffusion
 WORKDIR /stable-diffusion
 
 RUN <<EOF
-git reset --hard 2b7e3abe57963d199f1d825ddef87ae154c81045
+git reset --hard 5c31feb3a1096d437c94b6e1c3224eb7a7224a85
 git config --global http.postBuffer 1048576000
-ln -sf environments-and-requirements/requirements-lin-cuda.txt requirements.txt
-pip install -r requirements.txt
+pip install -r binary_installer/py3.10-linux-x86_64-cuda-reqs.txt
 EOF
 
 
-ARG BRANCH=development SHA=a9aa4e45aa6f5d5a2aa385349131d8733dd380fa
+# patch match:
+# https://github.com/invoke-ai/InvokeAI/blob/main/docs/installation/INSTALL_PATCHMATCH.md
+RUN <<EOF
+apt-get update
+# apt-get install build-essential python3-opencv libopencv-dev -y
+apt-get install make g++ libopencv-dev -y
+apt-get clean
+cd /usr/lib/x86_64-linux-gnu/pkgconfig/
+ln -sf opencv4.pc opencv.pc
+EOF
+
+ARG BRANCH=main SHA=5c31feb3a1096d437c94b6e1c3224eb7a7224a85
 RUN <<EOF
 git fetch
 git reset --hard
 git checkout ${BRANCH}
 git reset --hard ${SHA}
-pip install -r requirements.txt
+pip install -r binary_installer/py3.10-linux-x86_64-cuda-reqs.txt
 EOF
 
-RUN pip install --force-reinstall opencv-python-headless==4.5.5.64
+RUN pip install --force-reinstall opencv-python-headless && python3 -c "from patchmatch import patch_match"
+
 
 COPY . /docker/
 RUN <<EOF
@@ -41,9 +51,9 @@ touch ~/.invokeai
 EOF
 
 
-ENV ROOT=/stable-diffusion PRELOAD=false CLI_ARGS=""
+ENV ROOT=/stable-diffusion PYTHONPATH="${PYTHONPATH}:${ROOT}" PRELOAD=false CLI_ARGS=""
 EXPOSE 7860
 
 
 ENTRYPOINT ["/docker/entrypoint.sh"]
-CMD python3 -u scripts/invoke.py --web --host 0.0.0.0 --port 7860 --config /docker/models.yaml --root_dir . --outdir /output ${CLI_ARGS}
+CMD python3 -u scripts/invoke.py --web --host 0.0.0.0 --port 7860 --config /docker/models.yaml --root_dir ${ROOT} --outdir /output ${CLI_ARGS}
diff --git a/services/lstein/entrypoint.sh b/services/lstein/entrypoint.sh
index 3faeaf1..535f4f2 100755
--- a/services/lstein/entrypoint.sh
+++ b/services/lstein/entrypoint.sh
@@ -21,10 +21,8 @@ MOUNTS["${ROOT}/models/CompVis/stable-diffusion-safety-checker"]=/data/.cache/hu
 
 
 # hacks
-MOUNTS["/opt/conda/lib/python3.10/site-packages/facexlib/weights"]=/data/.cache/
 MOUNTS["${ROOT}/models/clipseg"]=/data/.cache/invoke/clipseg/
 
-
 for to_path in "${!MOUNTS[@]}"; do
   set -Eeuo pipefail
   from_path="${MOUNTS[${to_path}]}"
@@ -40,7 +38,8 @@ for to_path in "${!MOUNTS[@]}"; do
 done
 
 if "${PRELOAD}" == "true"; then
-  python3 -u scripts/preload_models.py --no-interactive --root . --config_file /docker/models.yaml
+  set -Eeuo pipefail
+  python3 -u scripts/preload_models.py --no-interactive --root ${ROOT} --config_file /docker/models.yaml
 fi
 
 exec "$@"