From c99a240ca59d727e0ca5b1ada9b736967827703c Mon Sep 17 00:00:00 2001
From: "Steven Lu (5950X 20.04[tmux])" <stevenlu443@gmail.com>
Date: Fri, 14 Jul 2023 07:22:13 -0400
Subject: [PATCH] seems to work

---
 services/comfy/Dockerfile             | 5 +++++
 services/comfy/extra_model_paths.yaml | 1 -
 2 files changed, 5 insertions(+), 1 deletion(-)

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