From 53b0e012a4fcb776e3343d2163e40993b150df3f Mon Sep 17 00:00:00 2001 From: zzzweakman Date: Mon, 31 Mar 2025 22:21:48 +0800 Subject: [PATCH] fix: unet config --- README.md | 5 +++++ inference.sh | 8 ++++++-- 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 4a281d5..3ad7f08 100644 --- a/README.md +++ b/README.md @@ -174,6 +174,11 @@ export FFMPEG_PATH=/musetalk/ffmpeg-4.4-amd64-static You can download weights manually as follows: 1. Download our trained [weights](https://huggingface.co/TMElyralab/MuseTalk). +```bash +# !pip install -U "huggingface_hub[cli]" +export HF_ENDPOINT=https://hf-mirror.com +huggingface-cli download TMElyralab/MuseTalk --local-dir models/ +``` 2. Download the weights of other components: - [sd-vae-ft-mse](https://huggingface.co/stabilityai/sd-vae-ft-mse) diff --git a/inference.sh b/inference.sh index e2d6f4b..d7d8426 100644 --- a/inference.sh +++ b/inference.sh @@ -20,9 +20,11 @@ config_path="./configs/inference/test.yaml" if [ "$version" = "v1.0" ]; then model_dir="./models/musetalk" unet_model_path="$model_dir/pytorch_model.bin" + unet_config="$model_dir/musetalk.json" elif [ "$version" = "v1.5" ]; then model_dir="./models/musetalkV15" unet_model_path="$model_dir/unet.pth" + unet_config="$model_dir/musetalk.json" else echo "Invalid version specified. Please use v1.0 or v1.5." exit 1 @@ -33,10 +35,12 @@ if [ "$version" = "v1.0" ]; then python3 -m scripts.inference \ --inference_config "$config_path" \ --result_dir "./results/test" \ - --unet_model_path "$unet_model_path" + --unet_model_path "$unet_model_path" \ + --unet_config "$unet_config" elif [ "$version" = "v1.5" ]; then python3 -m scripts.inference_alpha \ --inference_config "$config_path" \ --result_dir "./results/test" \ - --unet_model_path "$unet_model_path" + --unet_model_path "$unet_model_path" \ + --unet_config "$unet_config" fi \ No newline at end of file