test: fit Qwen2.5 online expansion on its declared two GPUs

Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
tzhouam committed 2026-10-03 22:06:27 +08:00
1 parent 864bc7b22a
commit 0b6591ae0a
1 file changed
+8 -1
@@ -21,7 +21,14 @@ models = ["Qwen/Qwen2.5-Omni-7B"]
# Single CI deploy YAML; rocm/xpu deltas are picked automatically via the
# platforms: section in vllm_omni/deploy/ci/qwen2_5_omni.yaml.
stage_configs = [modify_stage_config(get_deploy_config_path("ci/qwen2_5_omni.yaml"))]
stage_configs = [
modify_stage_config(
get_deploy_config_path("ci/qwen2_5_omni.yaml"),
# This suite requests two cards. The general CI overlay puts Code2Wav
# on a third card; share the Talker's card within the existing budgets.
updates={"stages": {2: {"devices": "1"}}},
)
]
# Create parameter combinations for model and stage config
test_params = [