From ee10e058cc94709cc5e605652ed2dd62f4d3b345 Mon Sep 17 00:00:00 2001
From: Self Denial <selfdenial@pm.me>
Date: Tue, 29 Oct 2024 15:38:49 -0600
Subject: [PATCH] Fix *upscale_models* path for `USE_KRITA=true`

---
 services/comfy/entrypoint.sh | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/services/comfy/entrypoint.sh b/services/comfy/entrypoint.sh
index 53cb6ce..309bada 100755
--- a/services/comfy/entrypoint.sh
+++ b/services/comfy/entrypoint.sh
@@ -27,7 +27,11 @@ done
 if [ "${USE_KRITA}" = "true" ]; then
   if [ "${KRITA_DOWNLOAD_MODELS:-false}" = "true" ]; then
     cd "${ROOT}/krita-ai-diffusion/scripts" && python scripts/download_models.py --recommended /data && cd ..
-    cd "${ROOT}/models/" mv -v upscale_models upscale_models.stock && ln -sT /data/models/upscale_models upscale_models
+  fi
+  [ -d "${ROOT}/models/upscale_models" ] && mv -v "${ROOT}/models/upscale_models" "${ROOT}/models/upscale_models.stock"
+  if [ ! -L "${ROOT}/models/upscale_models" ]; then
+    cd "${ROOT}/models"
+    ln -sfT /data/models/upscale_models upscale_models
   fi
   export USE_CNAUX="true" USE_IPAPLUS="true" USE_INPAINT="true"; USE_TOOLING="true"
 fi