From d11e071b9bf8e48fac228382baecfc8f5bc3de38 Mon Sep 17 00:00:00 2001
From: Mou Lai <geeklai@foxmail.com>
Date: Thu, 13 Oct 2022 21:39:23 +0800
Subject: [PATCH] Fix&Update to use lstein (PRELOAD=true)

1. Update docker-compose.yml: use PRELOAD=true for lstein.
2. Fix services/lstein/mount.sh: change "CodeFormer" to "Codeformer".
3. Update services/lstein/mount.sh: avoid having to re-download the data 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..9cfebd7 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/GFPGAN/weights
 
 for to_path in "${!MOUNTS[@]}"; do
   set -Eeuo pipefail