diff --git a/cu121-megapak/runner-scripts/download-models.txt b/cu121-megapak/runner-scripts/download-models.txt
index a0cdf9f..f24541b 100644
--- a/cu121-megapak/runner-scripts/download-models.txt
+++ b/cu121-megapak/runner-scripts/download-models.txt
@@ -9,12 +9,6 @@ https://huggingface.co/playgroundai/playground-v2.5-1024px-aesthetic/resolve/mai
 https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors
   dir=vae
   out=vae-ft-mse-840000-ema-pruned.safetensors
-https://huggingface.co/madebyollin/taesd/resolve/main/taesd_decoder.safetensors
-  dir=vae_approx
-  out=taesd_decoder.safetensors
-https://huggingface.co/madebyollin/taesdxl/resolve/main/taesdxl_decoder.safetensors
-  dir=vae_approx
-  out=taesdxl_decoder.safetensors
 
 # Upscale
 
diff --git a/cu121-megapak/runner-scripts/download.sh b/cu121-megapak/runner-scripts/download.sh
index 7e61530..607c7d2 100644
--- a/cu121-megapak/runner-scripts/download.sh
+++ b/cu121-megapak/runner-scripts/download.sh
@@ -92,6 +92,12 @@ echo "########################################"
 echo "[INFO] Downloading Models..."
 echo "########################################"
 
+# TAESD
+cd /root/ComfyUI/models/vae_approx
+clone_or_pull https://github.com/madebyollin/taesd.git
+cp taesd/*.pth ./
+rm -rf taesd
+
 # Models
 cd /root/ComfyUI/models
 aria2c \
diff --git a/cu121/scripts/download.sh b/cu121/scripts/download.sh
index 90f3356..129d6d8 100644
--- a/cu121/scripts/download.sh
+++ b/cu121/scripts/download.sh
@@ -29,6 +29,12 @@ echo "########################################"
 echo "[INFO] Downloading Models..."
 echo "########################################"
 
+# TAESD
+cd /home/runner/ComfyUI/models/vae_approx
+git clone --depth=1 --no-tags https://github.com/madebyollin/taesd.git
+cp taesd/*.pth ./
+rm -rf taesd
+
 # Models
 cd /home/runner/ComfyUI
 aria2c --input-file=/home/scripts/download.txt \
diff --git a/cu121/scripts/download.txt b/cu121/scripts/download.txt
index 2695ce6..770c6d8 100644
--- a/cu121/scripts/download.txt
+++ b/cu121/scripts/download.txt
@@ -12,12 +12,6 @@ https://huggingface.co/digiplay/Photon_v1/resolve/main/photon_v1.safetensors
 https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors
   dir=models/vae
   out=vae-ft-mse-840000-ema-pruned.safetensors
-https://huggingface.co/madebyollin/taesd/resolve/main/taesd_decoder.safetensors
-  dir=models/vae_approx
-  out=taesd_decoder.safetensors
-https://huggingface.co/madebyollin/taesdxl/resolve/main/taesdxl_decoder.safetensors
-  dir=models/vae_approx
-  out=taesdxl_decoder.safetensors
 
 # Upscale
 
diff --git a/cu124-megapak/runner-scripts/download-models.txt b/cu124-megapak/runner-scripts/download-models.txt
index c2c66fd..f30a6ff 100644
--- a/cu124-megapak/runner-scripts/download-models.txt
+++ b/cu124-megapak/runner-scripts/download-models.txt
@@ -9,12 +9,6 @@ https://huggingface.co/Comfy-Org/flux1-schnell/resolve/main/flux1-schnell-fp8.sa
 https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors
   dir=vae
   out=vae-ft-mse-840000-ema-pruned.safetensors
-https://huggingface.co/madebyollin/taesd/resolve/main/taesd_decoder.safetensors
-  dir=vae_approx
-  out=taesd_decoder.safetensors
-https://huggingface.co/madebyollin/taesdxl/resolve/main/taesdxl_decoder.safetensors
-  dir=vae_approx
-  out=taesdxl_decoder.safetensors
 
 # Upscale
 
diff --git a/cu124-megapak/runner-scripts/download.sh b/cu124-megapak/runner-scripts/download.sh
index f608a8b..0a09e41 100644
--- a/cu124-megapak/runner-scripts/download.sh
+++ b/cu124-megapak/runner-scripts/download.sh
@@ -90,6 +90,12 @@ echo "########################################"
 echo "[INFO] Downloading Models..."
 echo "########################################"
 
+# TAESD
+cd /root/ComfyUI/models/vae_approx
+clone_or_pull https://github.com/madebyollin/taesd.git
+cp taesd/*.pth ./
+rm -rf taesd
+
 # Models
 cd /root/ComfyUI/models
 aria2c \
diff --git a/rocm/runner-scripts/download.sh b/rocm/runner-scripts/download.sh
index eb856ed..26cba91 100644
--- a/rocm/runner-scripts/download.sh
+++ b/rocm/runner-scripts/download.sh
@@ -66,6 +66,12 @@ echo "########################################"
 echo "[INFO] Downloading Models..."
 echo "########################################"
 
+# TAESD
+cd /root/ComfyUI/models/vae_approx
+clone_or_pull https://github.com/madebyollin/taesd.git
+cp taesd/*.pth ./
+rm -rf taesd
+
 # Models
 cd /root/ComfyUI
 aria2c \
diff --git a/rocm/runner-scripts/download.txt b/rocm/runner-scripts/download.txt
index 7c08bd6..f1cc6c8 100644
--- a/rocm/runner-scripts/download.txt
+++ b/rocm/runner-scripts/download.txt
@@ -9,12 +9,6 @@ https://huggingface.co/playgroundai/playground-v2.5-1024px-aesthetic/resolve/mai
 https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors
   dir=models/vae
   out=vae-ft-mse-840000-ema-pruned.safetensors
-https://huggingface.co/madebyollin/taesd/resolve/main/taesd_decoder.safetensors
-  dir=models/vae_approx
-  out=taesd_decoder.safetensors
-https://huggingface.co/madebyollin/taesdxl/resolve/main/taesdxl_decoder.safetensors
-  dir=models/vae_approx
-  out=taesdxl_decoder.safetensors
 
 # Upscale