Jelajahi Sumber

Add ClipEncoder (#458)

https://github.com/AbdBarho/stable-diffusion-webui-docker/pull/432#issuecomment-1537380951
AbdBarho 3 tahun lalu
induk
melakukan
c836f41921

+ 1 - 0
data/.gitignore

@@ -25,3 +25,4 @@
 /ModelScope
 /LyCORIS
 /GLIGEN
+/CLIPEncoder

+ 1 - 1
docker-compose.yml

@@ -74,7 +74,7 @@ services:
     <<: *base_service
     profiles: ["comfy"]
     build: ./services/comfy/
-    image: sd-comfy:1
+    image: sd-comfy:2
     tty: true
     environment:
       - CLI_ARGS=

+ 2 - 1
services/comfy/extra_model_paths.yaml

@@ -14,10 +14,11 @@ a111:
   hypernetworks: Hypernetworks
   controlnet: ControlNet
   gligen: GLIGEN
+  clip: CLIPEncoder
+
   custom_nodes: config/comfy/custom_nodes
 
   # TODO: I am unsure about these, need more testing
-  # clip: .cache/clip
   # style_models: config/comfy/style_models
   # t2i_adapter: config/comfy/t2i_adapter
   # clip_vision: config/comfy/clip_vision

+ 21 - 1
services/download/download.sh

@@ -3,7 +3,27 @@
 set -Eeuo pipefail
 
 # TODO: maybe just use the .gitignore file to create all of these
-mkdir -vp /data/.cache /data/StableDiffusion /data/LyCORIS /data/Codeformer /data/ModelScope /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 /data/GLIGEN
+mkdir -vp /data/.cache \
+  /data/StableDiffusion \
+  /data/LyCORIS \
+  /data/Codeformer \
+  /data/ModelScope \
+  /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 \
+  /data/GLIGEN \
+  /data/CLIPEncoder # this is different from the cached pre-train ViT saved in /data/.cache/clip TODO: find out how
 
 echo "Downloading, this might take a while..."