test: fit Qwen2.5 online expansion on its declared two GPUs
Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
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 = [
|
||||
|
||||
Reference in new issue
Block a user