Adding mount points to shared models

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}

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