[CI] Fix Qwen3 Omni DSpark load test config (#56600)
Signed-off-by: Kevin Luu <51931015+khluu@users.noreply.github.com> Co-authored-by: OpenAI Codex <codex@openai.com>
This commit is contained in:
co-authored by
OpenAI Codex
parent
7bcca16729
commit
fd7cd4b883
@@ -354,6 +354,7 @@ def test_dspark_shares_target_embedding_with_smaller_draft_vocabulary():
|
||||
parallel_config=SimpleNamespace(tensor_parallel_size=1),
|
||||
attention_config=SimpleNamespace(backend=None),
|
||||
cache_config=SimpleNamespace(),
|
||||
load_config=SimpleNamespace(),
|
||||
model_config=SimpleNamespace(get_vocab_size=Mock(return_value=100)),
|
||||
)
|
||||
|
||||
@@ -368,6 +369,10 @@ def test_dspark_shares_target_embedding_with_smaller_draft_vocabulary():
|
||||
"vllm.v1.worker.gpu.spec_decode.eagle.utils.get_pp_group",
|
||||
return_value=SimpleNamespace(world_size=1),
|
||||
),
|
||||
patch(
|
||||
"vllm.v1.worker.gpu.spec_decode.utils.get_pp_group",
|
||||
return_value=SimpleNamespace(world_size=1),
|
||||
),
|
||||
patch(
|
||||
"vllm.compilation.backends.set_model_tag",
|
||||
return_value=nullcontext(),
|
||||
|
||||
Reference in New Issue
Block a user