Преглед изворни кода

Add mounts for `openpose` (#387)

Upon enabling the ControlNet addon from
https://github.com/AbdBarho/stable-diffusion-webui-docker/pull/385 one
might want to use the `openpose` preprocessors. Those are downloaded by
the addon the first time they are used. Without proper mounts those
networks will be downloaded on usage after each container start.
This PR enables those mounts to reduce data traffic.
Simon Oelerich пре 3 година
родитељ
комит
5d379bf7bc
4 измењених фајлова са 4 додато и 2 уклоњено
  1. 1 0
      data/.gitignore
  2. 1 1
      docker-compose.yml
  3. 1 0
      services/AUTOMATIC1111/entrypoint.sh
  4. 1 1
      services/download/download.sh

+ 1 - 0
data/.gitignore

@@ -21,3 +21,4 @@
 /embeddings
 /embeddings
 /Lora
 /Lora
 /ControlNet
 /ControlNet
+/openpose

+ 1 - 1
docker-compose.yml

@@ -28,7 +28,7 @@ services:
     <<: *base_service
     <<: *base_service
     profiles: ["auto"]
     profiles: ["auto"]
     build: ./services/AUTOMATIC1111
     build: ./services/AUTOMATIC1111
-    image: sd-auto:50
+    image: sd-auto:51
     environment:
     environment:
       - CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
       - CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
 
 

+ 1 - 0
services/AUTOMATIC1111/entrypoint.sh

@@ -36,6 +36,7 @@ MOUNTS["${ROOT}/models/BLIP"]="/data/BLIP"
 MOUNTS["${ROOT}/models/midas"]="/data/MiDaS"
 MOUNTS["${ROOT}/models/midas"]="/data/MiDaS"
 MOUNTS["${ROOT}/models/Lora"]="/data/Lora"
 MOUNTS["${ROOT}/models/Lora"]="/data/Lora"
 MOUNTS["${ROOT}/models/ControlNet"]="/data/ControlNet"
 MOUNTS["${ROOT}/models/ControlNet"]="/data/ControlNet"
+MOUNTS["${ROOT}/models/openpose"]="/data/openpose"
 
 
 MOUNTS["${ROOT}/embeddings"]="/data/embeddings"
 MOUNTS["${ROOT}/embeddings"]="/data/embeddings"
 MOUNTS["${ROOT}/config.json"]="/data/config/auto/config.json"
 MOUNTS["${ROOT}/config.json"]="/data/config/auto/config.json"

+ 1 - 1
services/download/download.sh

@@ -3,7 +3,7 @@
 set -Eeuo pipefail
 set -Eeuo pipefail
 
 
 # TODO: maybe just use the .gitignore file to create all of these
 # TODO: maybe just use the .gitignore file to create all of these
-mkdir -vp /data/.cache /data/StableDiffusion /data/Codeformer /data/GFPGAN /data/ESRGAN /data/BSRGAN /data/RealESRGAN /data/SwinIR /data/LDSR /data/ScuNET /data/embeddings /data/VAE /data/Deepdanbooru /data/MiDaS /data/Lora /data/ControlNet
+mkdir -vp /data/.cache /data/StableDiffusion /data/Codeformer /data/GFPGAN /data/ESRGAN /data/BSRGAN /data/RealESRGAN /data/SwinIR /data/LDSR /data/ScuNET /data/embeddings /data/VAE /data/Deepdanbooru /data/MiDaS /data/Lora /data/ControlNet /data/openpose
 
 
 echo "Downloading, this might take a while..."
 echo "Downloading, this might take a while..."