[BugFix] Remove stage_configs_path validation (#2741)

Signed-off-by: amy-why-3459 <wuhaiyan17@huawei.com>
This commit is contained in:
amy-why-3459 authored and GitHub committed 2026-04-14 17:32:00 +12:00
1 parent 8d23549b29
commit a5b38b5d0d
2 files changed
-12

No files matched your search

-7
View File
@@ -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(
-5
View File
@@ -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()