Merge pull request #17 from sudowind/feature/20250626-remove-lfs

Feature/20250626 remove lfs

.gitattributes

@@ -1,3 +0,0 @@
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -1,3 +1,10 @@
 *.pyc
 __pycache__
 /results/
+
+lm.pb
+model_1.onnx
+model.pb
+*.msc
+lite_avatar_weights/
+*.mv

README.md

@@ -13,6 +13,16 @@ We recommend a python version = 3.10 and cuda version = 11.8. Then build environ
 ```shell
 pip install -r requirements.txt
 ```
+
+## Model Preparation
+```shell
+# for windows
+download_model.bat
+
+# for linux
+bash download_model.sh
+```
+
 ## Inference
 ```
 python lite_avatar.py --data_dir /path/to/sample_data --audio_file /path/to/audio.wav --result_dir /path/to/result

download_model.bat

@@ -0,0 +1,21 @@
+@echo off
+REM Download LiteAvatar model files using modelscope
+
+echo Downloading LiteAvatar model files...
+
+modelscope download --model HumanAIGC-Engineering/LiteAvatarGallery lite_avatar_weights/lm.pb lite_avatar_weights/model_1.onnx lite_avatar_weights/model.pb --local_dir ./
+if %errorlevel% neq 0 (
+    echo Error downloading lite_avatar_weights
+    pause
+    exit /b 1
+)
+
+@REM move file
+move lite_avatar_weights\lm.pb ./weights/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch/lm/
+move lite_avatar_weights\model_1.onnx ./weights/
+move lite_avatar_weights\model.pb ./weights/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch/
+
+@REM remove folder
+rmdir lite_avatar_weights
+
+echo All model files downloaded successfully!

download_model.sh

@@ -0,0 +1,16 @@
+#!/bin/bash
+
+# download model
+echo "Downloading LiteAvatar model files..."
+
+modelscope download --model HumanAIGC-Engineering/LiteAvatarGallery lite_avatar_weights/lm.pb lite_avatar_weights/model_1.onnx lite_avatar_weights/model.pb --local_dir ./
+
+# move file
+mv lite_avatar_weights/lm.pb ./weights/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch/lm/
+mv lite_avatar_weights/model_1.onnx ./weights/
+mv lite_avatar_weights/model.pb ./weights/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch/
+
+# remove folder
+rm -rf lite_avatar_weights
+
+echo "All model files downloaded successfully!"

weights/model_1.onnx

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:312f0213a23a1c17ecb66f9edf8413b8ef193c0d4d2f5f9dd6714a178492c34c
-size 184415825

weights/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch/lm/lm.pb

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5b59db5f607fde57bd8d6c8df8f22ce5b37ec7adcdb7b4ee785a4478f06c1c50
-size 237373385

weights/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch/model.pb

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5bba782a5e9196166233b9ab12ba04cadff9ef9212b4ff6153ed9290ff679025
-size 880502012