diff --git a/services/comfy/Dockerfile b/services/comfy/Dockerfile
index 003363a..7453c8e 100644
--- a/services/comfy/Dockerfile
+++ b/services/comfy/Dockerfile
@@ -22,6 +22,11 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 
 WORKDIR ${ROOT}
 
+ADD https://github.com/madebyollin/taesd/raw/main/taesd_decoder.pth \
+  ${ROOT}/models/vae_approx/taesd_decoder.pth
+ADD https://github.com/madebyollin/taesd/raw/main/taesdxl_decoder.pth \
+  ${ROOT}/models/vae_approx/taesdxl_decoder.pth
+
 # add info
 COPY . /docker/
 RUN cp /docker/extra_model_paths.yaml ${ROOT}
diff --git a/services/comfy/extra_model_paths.yaml b/services/comfy/extra_model_paths.yaml
index e4f6886..eb374eb 100644
--- a/services/comfy/extra_model_paths.yaml
+++ b/services/comfy/extra_model_paths.yaml
@@ -15,7 +15,6 @@ a111:
   gligen: models/GLIGEN
   clip: models/CLIPEncoder
   embeddings: embeddings
-  vae_approx: models/vae_approx
 
   custom_nodes: config/comfy/custom_nodes