From 402c691a4929d52b7ff287c0d6aedd89db9429b4 Mon Sep 17 00:00:00 2001
From: Mou Lai <geeklai@foxmail.com>
Date: Fri, 14 Oct 2022 21:52:01 +0800
Subject: [PATCH] Fix & Update to use lstein (#131)

1. Update `docker-compose.yml`: use `PRELOAD=true` for lstein. See the
[note](https://github.com/invoke-ai/InvokeAI/blob/94bad8555a8832eca01ed7da6797f65cf6bb9cfb/docs/installation/INSTALL_LINUX.md?plain=1#L60)
in the installation guide of InvokeAI.
3. Fix `services/lstein/mount.sh`: change `CodeFormer` to `Codeformer`.
4. Update `services/lstein/mount.sh`: avoid having to re-download the
`${PWD}/gfpgan/weights/...` every time the container is started.
---
 docker-compose.yml       | 2 +-
 services/lstein/mount.sh | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 2674e83..3f1e392 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -53,5 +53,5 @@ services:
     profiles: ["lstein"]
     build: ./services/lstein/
     environment:
-      - PRELOAD=false
+      - PRELOAD=true
       - CLI_ARGS=
diff --git a/services/lstein/mount.sh b/services/lstein/mount.sh
index c8cd0fc..45bf13e 100755
--- a/services/lstein/mount.sh
+++ b/services/lstein/mount.sh
@@ -9,10 +9,11 @@ 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/invoke/restoration/codeformer/weights"]=/data/CodeFormer
+MOUNTS["${PWD}/ldm/invoke/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
+MOUNTS["${PWD}/gfpgan/weights"]=/data/.cache
 
 for to_path in "${!MOUNTS[@]}"; do
   set -Eeuo pipefail