'cu124-cn' Update TAESD

YanWenKun · Sep 9, 2024 · cb7be58 · cb7be58
1 parent 02fb3a2
commit cb7be58
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/cu124-cn/runner-scripts/download-models.txt b/cu124-cn/runner-scripts/download-models.txt
@@ -10,6 +10,20 @@ https://hf-mirror.com/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse
   dir=vae
   out=vae-ft-mse-840000-ema-pruned.safetensors
 
+# TAESD（用于高质量预览）
+https://mirror.ghproxy.com/https://raw.githubusercontent.com/madebyollin/taesd/main/taesd_decoder.pth
+  dir=vae_approx
+  out=taesd_decoder.pth
+https://mirror.ghproxy.com/https://raw.githubusercontent.com/madebyollin/taesd/main/taesdxl_decoder.pth
+  dir=vae_approx
+  out=taesdxl_decoder.pth
+https://mirror.ghproxy.com/https://raw.githubusercontent.com/madebyollin/taesd/main/taesd3_decoder.pth
+  dir=vae_approx
+  out=taesd3_decoder.pth
+https://mirror.ghproxy.com/https://raw.githubusercontent.com/madebyollin/taesd/main/taef1_decoder.pth
+  dir=vae_approx
+  out=taef1_decoder.pth
+
 # 放大
 
 https://hf-mirror.com/gemasai/4x_NMKD-Siax_200k/resolve/main/4x_NMKD-Siax_200k.pth

diff --git a/cu124-cn/runner-scripts/download.sh b/cu124-cn/runner-scripts/download.sh
@@ -92,13 +92,6 @@ echo "########################################"
 echo "[INFO] 下载模型……"
 echo "########################################"
 
-# TAESD（用于高质量预览）
-mkdir -p /root/ComfyUI/models/vae_approx
-cd /root/ComfyUI/models/vae_approx
-clone_or_pull https://mirror.ghproxy.com/https://github.com/madebyollin/taesd.git
-cp taesd/*.pth ./
-rm -rf taesd
-
 cd /root/ComfyUI/models
 aria2c \
   --input-file=/runner-scripts/download-models.txt \