From e228748b23cc9b7aa2577705cdee0ad39d42e53b Mon Sep 17 00:00:00 2001 From: amy-why-3459 Date: Mon, 13 Apr 2026 22:50:29 +0800 Subject: [PATCH] Remove stage_configs_path validation Signed-off-by: amy-why-3459 --- tests/engine/test_arg_utils.py | 7 ------- vllm_omni/engine/arg_utils.py | 5 ----- 2 files changed, 12 deletions(-) diff --git a/tests/engine/test_arg_utils.py b/tests/engine/test_arg_utils.py index a1fc18f8456..35d55f1cc4e 100644 --- a/tests/engine/test_arg_utils.py +++ b/tests/engine/test_arg_utils.py @@ -118,13 +118,6 @@ def test_qwen3_tts_codec_frame_rate_patching(): assert omni_config.codec_frame_rate_hz == 12.3 -def test_stage_configs_path_blocks_create_model_config(): - """create_model_config() should raise when stage_configs_path is set.""" - args = OmniEngineArgs(stage_configs_path="/some/path.yaml") - with pytest.raises(RuntimeError, match="stage_configs_path"): - args.create_model_config() - - def test_from_cli_args_picks_up_stage_configs_path(): """from_cli_args should pick up stage_configs_path from namespace.""" ns = argparse.Namespace( diff --git a/vllm_omni/engine/arg_utils.py b/vllm_omni/engine/arg_utils.py index 4e2ad9b257c..d61102c7e13 100644 --- a/vllm_omni/engine/arg_utils.py +++ b/vllm_omni/engine/arg_utils.py @@ -194,11 +194,6 @@ def create_model_config(self) -> OmniModelConfig: Returns: OmniModelConfig instance with all configuration fields set """ - if self.stage_configs_path is not None: - raise RuntimeError( - "create_model_config() should not be called when stage_configs_path is set. " - "Per-stage model configs are resolved from the stage config YAML." - ) # register omni models to avoid model not found error self._ensure_omni_models_registered()