mirror of
https://github.com/TMElyralab/MuseTalk.git
synced 2026-02-04 09:29:20 +08:00
feat: update download_weights.bat (#372)
更换weights下载工具 更换face-parse-bisent源
This commit is contained in:
@@ -14,32 +14,28 @@ mkdir %CheckpointsDir%\sd-vae-ft-mse
|
|||||||
mkdir %CheckpointsDir%\whisper
|
mkdir %CheckpointsDir%\whisper
|
||||||
|
|
||||||
:: Install required packages
|
:: Install required packages
|
||||||
pip install -U "huggingface_hub[cli]"
|
pip install -U "huggingface_hub[hf_xet]"
|
||||||
pip install gdown
|
|
||||||
|
|
||||||
:: Set HuggingFace endpoint
|
:: Set HuggingFace endpoint
|
||||||
set HF_ENDPOINT=https://hf-mirror.com
|
set HF_ENDPOINT=https://hf-mirror.com
|
||||||
|
|
||||||
:: Download MuseTalk weights
|
:: Download MuseTalk weights
|
||||||
huggingface-cli download TMElyralab/MuseTalk --local-dir %CheckpointsDir%
|
hf download TMElyralab/MuseTalk --local-dir %CheckpointsDir%
|
||||||
|
|
||||||
:: Download SD VAE weights
|
:: 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
|
:: 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
|
:: 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
|
:: 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)
|
:: Download face-parse-bisent weights
|
||||||
gdown --id 154JgKpzCPW82qINcVieuPH3fZ2e0P812 -O %CheckpointsDir%\face-parse-bisent\79999_iter.pth
|
hf download ManyOtherFunctions/face-parse-bisent --local-dir %CheckpointsDir%\face-parse-bisent --include "79999_iter.pth" "resnet18-5c106cde.pth"
|
||||||
|
|
||||||
:: Download ResNet weights
|
|
||||||
curl -L https://download.pytorch.org/models/resnet18-5c106cde.pth -o %CheckpointsDir%\face-parse-bisent\resnet18-5c106cde.pth
|
|
||||||
|
|
||||||
echo All weights have been downloaded successfully!
|
echo All weights have been downloaded successfully!
|
||||||
endlocal
|
endlocal
|
||||||
|
|||||||
Reference in New Issue
Block a user