AbdBarho/stable-diffusion-webui-docker
1
0
Fork 0
mirror of https://github.com/AbdBarho/stable-diffusion-webui-docker.git synced 2026-02-04 06:34:19 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix&Update to use lstein (PRELOAD=true)

Browse source 
1. Update docker-compose.yml: use PRELOAD=true for lstein.
2. Fix services/lstein/mount.sh: change "CodeFormer" to "Codeformer".
3. Update services/lstein/mount.sh: avoid having to re-download the data every time the container is started.
This commit is contained in:
Mou Lai 2022-10-13 21:39:23 +08:00
parent b60c787474
commit d11e071b9b
2 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
docker-compose.yml
View file

@ -53,5 +53,5 @@ services:
profiles: ["lstein"] profiles: ["lstein"]
build: ./services/lstein/ build: ./services/lstein/
environment: environment:
- PRELOAD=false - PRELOAD=true
- CLI_ARGS= - CLI_ARGS=

3
services/lstein/mount.sh
View file

@ -9,10 +9,11 @@ MOUNTS["/root/.cache"]=/data/.cache
# ui specific # ui specific
MOUNTS["${PWD}/models/ldm/stable-diffusion-v1/model.ckpt"]=/data/StableDiffusion/model.ckpt MOUNTS["${PWD}/models/ldm/stable-diffusion-v1/model.ckpt"]=/data/StableDiffusion/model.ckpt
MOUNTS["${PWD}/src/gfpgan/experiments/pretrained_models/GFPGANv1.4.pth"]=/data/GFPGAN/GFPGANv1.4.pth MOUNTS["${PWD}/src/gfpgan/experiments/pretrained_models/GFPGANv1.4.pth"]=/data/GFPGAN/GFPGANv1.4.pth
MOUNTS["${PWD}/ldm/invoke/restoration/codeformer/weights"]=/data/CodeFormer MOUNTS["${PWD}/ldm/invoke/restoration/codeformer/weights"]=/data/Codeformer
# hacks # hacks
MOUNTS["/opt/conda/lib/python3.9/site-packages/facexlib/weights"]=/data/.cache MOUNTS["/opt/conda/lib/python3.9/site-packages/facexlib/weights"]=/data/.cache
MOUNTS["/opt/conda/lib/python3.9/site-packages/realesrgan/weights"]=/data/RealESRGAN MOUNTS["/opt/conda/lib/python3.9/site-packages/realesrgan/weights"]=/data/RealESRGAN
MOUNTS["${PWD}/gfpgan/weights"]=/data/GFPGAN/weights
for to_path in "${!MOUNTS[@]}"; do for to_path in "${!MOUNTS[@]}"; do
set -Eeuo pipefail set -Eeuo pipefail