From 0a89dec45a0192b824e3cf4daf96c239440c5ed8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E6=B4=8B?= Date: Fri, 26 Sep 2025 13:44:17 +0800 Subject: [PATCH] feat: update download_weights.bat (#372) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 更换weights下载工具 更换face-parse-bisent源 --- download_weights.bat | 22 +++++++++------------- 1 file changed, 9 insertions(+), 13 deletions(-) diff --git a/download_weights.bat b/download_weights.bat index c6fbdd2..2407f52 100644 --- a/download_weights.bat +++ b/download_weights.bat @@ -14,32 +14,28 @@ mkdir %CheckpointsDir%\sd-vae-ft-mse mkdir %CheckpointsDir%\whisper :: Install required packages -pip install -U "huggingface_hub[cli]" -pip install gdown +pip install -U "huggingface_hub[hf_xet]" :: Set HuggingFace endpoint set HF_ENDPOINT=https://hf-mirror.com :: Download MuseTalk weights -huggingface-cli download TMElyralab/MuseTalk --local-dir %CheckpointsDir% +hf download TMElyralab/MuseTalk --local-dir %CheckpointsDir% :: Download SD VAE weights -huggingface-cli download stabilityai/sd-vae-ft-mse --local-dir %CheckpointsDir%\sd-vae --include "config.json" "diffusion_pytorch_model.bin" +hf download stabilityai/sd-vae-ft-mse --local-dir %CheckpointsDir%\sd-vae --include "config.json" "diffusion_pytorch_model.bin" :: Download Whisper weights -huggingface-cli download openai/whisper-tiny --local-dir %CheckpointsDir%\whisper --include "config.json" "pytorch_model.bin" "preprocessor_config.json" +hf download openai/whisper-tiny --local-dir %CheckpointsDir%\whisper --include "config.json" "pytorch_model.bin" "preprocessor_config.json" :: Download DWPose weights -huggingface-cli download yzd-v/DWPose --local-dir %CheckpointsDir%\dwpose --include "dw-ll_ucoco_384.pth" +hf download yzd-v/DWPose --local-dir %CheckpointsDir%\dwpose --include "dw-ll_ucoco_384.pth" :: Download SyncNet weights -huggingface-cli download ByteDance/LatentSync --local-dir %CheckpointsDir%\syncnet --include "latentsync_syncnet.pt" +hf download ByteDance/LatentSync --local-dir %CheckpointsDir%\syncnet --include "latentsync_syncnet.pt" -:: Download Face Parse Bisent weights (using gdown) -gdown --id 154JgKpzCPW82qINcVieuPH3fZ2e0P812 -O %CheckpointsDir%\face-parse-bisent\79999_iter.pth - -:: Download ResNet weights -curl -L https://download.pytorch.org/models/resnet18-5c106cde.pth -o %CheckpointsDir%\face-parse-bisent\resnet18-5c106cde.pth +:: Download face-parse-bisent weights +hf download ManyOtherFunctions/face-parse-bisent --local-dir %CheckpointsDir%\face-parse-bisent --include "79999_iter.pth" "resnet18-5c106cde.pth" echo All weights have been downloaded successfully! -endlocal \ No newline at end of file +endlocal