Skip to content

Commit 05ddcb9

Browse files
committed
test: add qwen3 eagle3 test case
Signed-off-by: LeChen <[email protected]>
1 parent 5872f06 commit 05ddcb9

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/v1/e2e/test_spec_decode.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,8 @@ def test_ngram_correctness(
125125

126126
@pytest.mark.parametrize(
127127
["model_setup", "mm_enabled"], [
128+
(("eagle3", "Qwen/Qwen3-8B",
129+
"AngelSlim/Qwen3-8B_eagle3", 1), False),
128130
(("eagle", "meta-llama/Llama-3.1-8B-Instruct",
129131
"yuhuili/EAGLE-LLaMA3.1-Instruct-8B", 1), False),
130132
(("eagle3", "meta-llama/Llama-3.1-8B-Instruct",
@@ -140,7 +142,7 @@ def test_ngram_correctness(
140142
True,
141143
marks=pytest.mark.skip(reason="Skipping due to CI OOM issues")),
142144
],
143-
ids=["llama3_eagle", "llama3_eagle3", "llama4_eagle", "llama4_eagle_mm"])
145+
ids=["qwen3_eagle3", "llama3_eagle", "llama3_eagle3", "llama4_eagle", "llama4_eagle_mm"])
144146
def test_eagle_correctness(
145147
monkeypatch: pytest.MonkeyPatch,
146148
sampling_config: SamplingParams,

0 commit comments

Comments
 (0)