[BugFix] Remove stage_configs_path validation (#2741)
Signed-off-by: amy-why-3459 <wuhaiyan17@huawei.com>
This commit is contained in:
1 parent
8d23549b29
commit
a5b38b5d0d
2 files changed
-12
No files matched your search
@@ -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(
|
||||
|
||||
@@ -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()
|
||||
|
||||
|
||||
Reference in new issue
Block a user