From 77cfe4965ee3ccf40634f1be3486da640497c2ca Mon Sep 17 00:00:00 2001
From: Paul Lamb <plamb@outlook.com>
Date: Sat, 30 Dec 2023 22:20:58 -0800
Subject: [PATCH] Adding mount points to shared models

---
 services/fooocus/Dockerfile    |  7 +++----
 services/fooocus/entrypoint.sh | 22 +++++++---------------
 2 files changed, 10 insertions(+), 19 deletions(-)

diff --git a/services/fooocus/Dockerfile b/services/fooocus/Dockerfile
index 5f6e120..7a1ff77 100644
--- a/services/fooocus/Dockerfile
+++ b/services/fooocus/Dockerfile
@@ -8,12 +8,12 @@ ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
 
 RUN apt-get update && apt-get install -y git && apt-get clean
 
-# add some packages for some custom nodes in comfyui
+# add in required packages
 RUN apt-get install 'libglib2.0-0'  -y
-RUN apt-get update
 RUN apt-get install 'libgl1-mesa-glx' -y
 RUN apt-get install 'python-dev' -y
 
+# set this to your target branch commit
 ARG BRANCH=main SHA=8e62a72a63b30a3067d1a1bc3f8d226824bd9283
 
 ENV ROOT=/stable-diffusion
@@ -39,10 +39,9 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 
 # add info
 COPY . /docker/
+RUN cp /docker/config.txt ${ROOT}
 RUN chmod u+x /docker/entrypoint.sh
 
-#ENV NVIDIA_VISIBLE_DEVICES=all
-#ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
 EXPOSE 7860
 ENTRYPOINT ["/docker/entrypoint.sh"]
 CMD python -u entry_with_update.py --listen --port 7860 ${CLI_ARGS}
diff --git a/services/fooocus/entrypoint.sh b/services/fooocus/entrypoint.sh
index 3cdc921..ee1416e 100644
--- a/services/fooocus/entrypoint.sh
+++ b/services/fooocus/entrypoint.sh
@@ -4,22 +4,14 @@ set -Eeuo pipefail
 
 declare -A MOUNTS
 
-#mkdir -p ${CONFIG_DIR} ${ROOT}/configs/stable-diffusion/
+MOUNTS["${ROOT}/outputs"]="/output/fooocus"
 
-# cache
-#MOUNTS["/root/.cache"]=/data/.cache/
-
-# this is really just a hack to avoid migrations
-#rm -rf ${HF_HOME}/diffusers
-
-# ui specific
-#MOUNTS["${ROOT}/models/codeformer"]=/data/models/Codeformer/
-#MOUNTS["${ROOT}/models/gfpgan/GFPGANv1.4.pth"]=/data/models/GFPGAN/GFPGANv1.4.pth
-#MOUNTS["${ROOT}/models/gfpgan/weights"]=/data/models/GFPGAN/
-#MOUNTS["${ROOT}/models/realesrgan"]=/data/models/RealESRGAN/
-#MOUNTS["${ROOT}/models/ldm"]=/data/.cache/invoke/ldm/
-
-# hacks
+# ui specific mounts
+MOUNTS["${ROOT}/models/checkpoints"]=/data/models/Stable-diffusion/
+MOUNTS["${ROOT}/models/loras"]=/data/models/Lora/
+MOUNTS["${ROOT}/models/embeddings"]=/data/models/embeddings/
+MOUNTS["${ROOT}/models/vae_approx"]=/data/models/VAE/
+MOUNTS["${ROOT}/models/upscale_models"]=/data/models/upscale_models/
 
 for to_path in "${!MOUNTS[@]}"; do
   set -Eeuo pipefail