From 3ee9adb7704e8bba0ee420f4e3a837fa6fce086f Mon Sep 17 00:00:00 2001
From: AbdBarho <ka70911@gmail.com>
Date: Mon, 26 Sep 2022 08:19:29 +0200
Subject: [PATCH] lstein

---
 services/lstein/Dockerfile |  2 +-
 services/lstein/mount.sh   | 11 +++++------
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/services/lstein/Dockerfile b/services/lstein/Dockerfile
index 5063747..8464c6c 100644
--- a/services/lstein/Dockerfile
+++ b/services/lstein/Dockerfile
@@ -23,7 +23,7 @@ conda clean -a -y
 EOF
 
 
-ARG BRANCH=development SHA=50d607ffea3734072a80e38b09ba0c3758af5d40
+ARG BRANCH=development SHA=b40bfb5116b7fc618f78a0d152005ceb46153443
 # this breaks on generation:
 # there is a new UI anyway, but it is not by any means ready.
 # ARG BRANCH=development SHA=bdbc76fcd4bd3362312dc91b087d9af66de423b1
diff --git a/services/lstein/mount.sh b/services/lstein/mount.sh
index bae7a0b..0b31bdf 100755
--- a/services/lstein/mount.sh
+++ b/services/lstein/mount.sh
@@ -7,12 +7,11 @@ ROOT=/stable-diffusion
 mkdir -p "${ROOT}/models/ldm/stable-diffusion-v1/"
 ln -sf /cache/models/model.ckpt "${ROOT}/models/ldm/stable-diffusion-v1/model.ckpt"
 
-if test -f /cache/models/GFPGANv1.3.pth; then
-  base="${ROOT}/src/gfpgan/experiments/pretrained_models/"
-  mkdir -p "${base}"
-  ln -sf /cache/models/GFPGANv1.3.pth "${base}/GFPGANv1.3.pth"
-  echo "Mounted GFPGANv1.3.pth"
-fi
+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