diff --git a/cli/podcast.py b/cli/podcast.py index d61dfd5..c5a5117 100644 --- a/cli/podcast.py +++ b/cli/podcast.py @@ -57,7 +57,7 @@ def run_inference( parser.add_argument("--seed", type=int, default=1988, help="Random seed") args = parser.parse_args() - with open(args.json_path, "r") as f: + with open(args.json_path, "r", encoding="utf-8") as f: data = json.load(f) inputs = podcast_format_parser(data) run_inference( diff --git a/example/infer_dialogue.bat b/example/infer_dialogue.bat new file mode 100644 index 0000000..206515d --- /dev/null +++ b/example/infer_dialogue.bat @@ -0,0 +1,21 @@ +@echo off +setlocal enabledelayedexpansion + +:: 获取当前批处理文件所在目录的父目录作为 PYTHONPATH +set "SCRIPT_DIR=%~dp0" +for %%I in ("%SCRIPT_DIR%\..") do set "PARENT_DIR=%%~fI" +set "PYTHONPATH=%PARENT_DIR%" +echo PYTHONPATH set to: %PYTHONPATH% + +:: 设置模型和输入路径 +set "model_dir=pretrained_models/SoulX-Podcast-1.7B" +set "input_file=example/podcast_script/script_mandarin.json" + +:: 运行 Python 脚本 +python cli/podcast.py ^ + --json_path "%input_file%" ^ + --model_path "%model_dir%" ^ + --output_path outputs/mandarin.wav ^ + --seed 7 + +endlocal \ No newline at end of file