mirror of
https://github.com/AbdBarho/stable-diffusion-webui-docker.git
synced 2024-08-17 14:30:27 +03:00
Adding mount points to shared models
This commit is contained in:
parent
18fb90ea22
commit
77cfe4965e
@ -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}
|
||||
|
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user