[Config] Add HunyuanImage3 deploy configs (#3172)

Signed-off-by: KexiongYu <yukexiong1@huawei.com>
Signed-off-by: Y. Fisher <yukexiong1@huawei.com>
This commit is contained in:
Y. Fisher authored and GitHub committed 2026-05-11 22:49:19 +08:00
1 parent c9a8556c24
commit a33e2eb588
26 files changed
+672 -742

No files matched your search

+39
View File
@@ -18,6 +18,7 @@ from vllm_omni.entrypoints.utils import (
_filter_dict_like_object,
coerce_param_message_types,
filter_dataclass_kwargs,
filter_stages,
load_and_resolve_stage_configs,
load_stage_configs_from_yaml,
resolve_model_config_path,
@@ -401,6 +402,44 @@ class TestLoadAndResolveStageConfigs:
assert stage_configs[1].runtime.num_replicas == 3
assert stage_configs[1].runtime.devices == "1,2,3"
def test_filter_stages_selects_mode_stages_without_mutating_stage_config(self, tmp_path):
config_path = tmp_path / "deploy.yaml"
config_path.write_text(
"""modes:
- mode: text-to-text
stages: [0]
- mode: text-to-image
stages: [0, 1]
""",
encoding="utf-8",
)
stages = [
create_config(
{
"stage_id": 0,
"runtime": {"requires_multimodal_data": True},
"final_output": False,
"final_output_type": None,
}
),
create_config(
{
"stage_id": 1,
"runtime": {"requires_multimodal_data": True},
"final_output": True,
"final_output_type": "image",
}
),
]
filtered = filter_stages(str(config_path), stages, {"mode": "text-to-text"})
assert len(filtered) == 1
assert filtered[0].stage_id == 0
assert filtered[0].runtime.requires_multimodal_data is True
assert filtered[0].final_output is False
assert filtered[0].final_output_type is None
class TestLoadStageConfigsFromYaml:
"""Regression tests for stage-config loading and merging."""