download all of TAESD

YanWenKun · Sep 9, 2024 · c2a96ac · c2a96ac
1 parent 84d3d6d
commit c2a96ac
Show file tree

Hide file tree

Showing 8 changed files with 24 additions and 24 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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