diff --git a/data/.gitignore b/data/.gitignore
index 3bb809d..90e640d 100644
--- a/data/.gitignore
+++ b/data/.gitignore
@@ -25,3 +25,4 @@
 /ModelScope
 /LyCORIS
 /GLIGEN
+/CLIPEncoder
diff --git a/docker-compose.yml b/docker-compose.yml
index 711cee8..a42425d 100644
--- a/docker-compose.yml
+++ b/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=
diff --git a/services/comfy/extra_model_paths.yaml b/services/comfy/extra_model_paths.yaml
index ba27ae2..d16dfef 100644
--- a/services/comfy/extra_model_paths.yaml
+++ b/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
diff --git a/services/download/download.sh b/services/download/download.sh
index 03835ac..68ea540 100755
--- a/services/download/download.sh
+++ b/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..."