From f8624db93a3832136189e7cc7fec57d9f5c6e076 Mon Sep 17 00:00:00 2001 From: boatman <109857087+sphinxkkkbc@users.noreply.github.com> Date: Fri, 8 May 2026 21:03:47 +0800 Subject: [PATCH] [BugFix]Fix default stage config path in voxcpm2 (#3447) Signed-off-by: sphinxkkkbc Co-authored-by: sphinxkkkbc --- .../text_to_speech/voxcpm2/end2end.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/examples/offline_inference/text_to_speech/voxcpm2/end2end.py b/examples/offline_inference/text_to_speech/voxcpm2/end2end.py index c9ac766fc..7d77d8173 100644 --- a/examples/offline_inference/text_to_speech/voxcpm2/end2end.py +++ b/examples/offline_inference/text_to_speech/voxcpm2/end2end.py @@ -18,7 +18,6 @@ from vllm_omni import Omni from vllm_omni.engine.arg_utils import nullify_stage_engine_defaults REPO_ROOT = Path(__file__).resolve().parents[4] -DEFAULT_STAGE_CONFIGS_PATH = str(REPO_ROOT / "vllm_omni" / "model_executor" / "stage_configs" / "voxcpm2.yaml") SAMPLE_RATE = 48_000 @@ -43,10 +42,11 @@ def parse_args(): help="Directory for output WAV files.", ) parser.add_argument( - "--stage-configs-path", + "--deploy-config", type=str, - default=DEFAULT_STAGE_CONFIGS_PATH, - help="Path to the stage config YAML file.", + default=None, + help="Override the deploy config path. If unset, auto-loads " + "vllm_omni/deploy/voxcpm2.yaml based on the HF model_type.", ) parser.add_argument( "--ref-audio", @@ -96,7 +96,7 @@ def main(): engine = Omni( model=args.model, - stage_configs_path=args.stage_configs_path, + deploy_config=args.deploy_config, ) from transformers import AutoTokenizer