diff --git a/Dockerfile b/Dockerfile
index 8ef0adf..4de670a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -117,6 +117,8 @@ RUN mkdir -p /runpod-volume/models/checkpoints \
     /runpod-volume/models/controlnet \
     /runpod-volume/models/clip \
     /runpod-volume/models/upscale_models \
+    /runpod-volume/models/diffusion_models \
+    /runpod-volume/models/text_encoders \
     /runpod-volume/.cache/huggingface \
     /workspace/.cache
 
@@ -126,7 +128,9 @@ RUN rm -rf models/checkpoints && ln -s /runpod-volume/models/checkpoints models/
     rm -rf models/vae && ln -s /runpod-volume/models/vae models/vae && \
     rm -rf models/controlnet && ln -s /runpod-volume/models/controlnet models/controlnet && \
     rm -rf models/clip && ln -s /runpod-volume/models/clip models/clip && \
-    rm -rf models/upscale_models && ln -s /runpod-volume/models/upscale_models models/upscale_models
+    rm -rf models/upscale_models && ln -s /runpod-volume/models/upscale_models models/upscale_models && \
+    rm -rf models/diffusion_models && ln -s /runpod-volume/models/diffusion_models models/diffusion_models && \
+    rm -rf models/text_encoders && ln -s /runpod-volume/models/text_encoders models/text_encoders
 
 # Symlink HuggingFace cache
 RUN ln -s /runpod-volume/.cache/huggingface /workspace/.cache/huggingface