From a5b38b5d0d612d4be0b452dfd29c552f2dfa94a3 Mon Sep 17 00:00:00 2001 From: amy-why-3459 Date: Tue, 14 Apr 2026 13:32:00 +0800 Subject: [PATCH] [BugFix] Remove stage_configs_path validation (#2741) 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 a1fc18f84..35d55f1cc 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 4e2ad9b25..d61102c7e 100644 --- a/vllm_omni/engine/arg_utils.py +++ b/vllm_omni/engine/arg_utils.py @@ -194,11 +194,6 @@ class OmniEngineArgs(EngineArgs): 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()