Skip to content

Commit 6a42c92

Browse files
committed
test: add qwen3 eagle3 test case
Signed-off-by: LeChen <[email protected]>
1 parent ddca7a5 commit 6a42c92

File tree

1 file changed

+21
-18
lines changed

1 file changed

+21
-18
lines changed

tests/v1/e2e/test_spec_decode.py

Lines changed: 21 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -123,24 +123,27 @@ def test_ngram_correctness(
123123
cleanup_dist_env_and_memory()
124124

125125

126-
@pytest.mark.parametrize(
127-
["model_setup", "mm_enabled"], [
128-
(("eagle", "meta-llama/Llama-3.1-8B-Instruct",
129-
"yuhuili/EAGLE-LLaMA3.1-Instruct-8B", 1), False),
130-
(("eagle3", "meta-llama/Llama-3.1-8B-Instruct",
131-
"yuhuili/EAGLE3-LLaMA3.1-Instruct-8B", 1), False),
132-
pytest.param(
133-
("eagle", "meta-llama/Llama-4-Scout-17B-16E-Instruct",
134-
"morgendave/EAGLE-Llama-4-Scout-17B-16E-Instruct", 4),
135-
False,
136-
marks=pytest.mark.skip(reason="Skipping due to CI OOM issues")),
137-
pytest.param(
138-
("eagle", "meta-llama/Llama-4-Scout-17B-16E-Instruct",
139-
"morgendave/EAGLE-Llama-4-Scout-17B-16E-Instruct", 4),
140-
True,
141-
marks=pytest.mark.skip(reason="Skipping due to CI OOM issues")),
142-
],
143-
ids=["llama3_eagle", "llama3_eagle3", "llama4_eagle", "llama4_eagle_mm"])
126+
@pytest.mark.parametrize(["model_setup", "mm_enabled"], [
127+
(("eagle3", "Qwen/Qwen3-8B", "AngelSlim/Qwen3-8B_eagle3", 1), False),
128+
(("eagle", "meta-llama/Llama-3.1-8B-Instruct",
129+
"yuhuili/EAGLE-LLaMA3.1-Instruct-8B", 1), False),
130+
(("eagle3", "meta-llama/Llama-3.1-8B-Instruct",
131+
"yuhuili/EAGLE3-LLaMA3.1-Instruct-8B", 1), False),
132+
pytest.param(
133+
("eagle", "meta-llama/Llama-4-Scout-17B-16E-Instruct",
134+
"morgendave/EAGLE-Llama-4-Scout-17B-16E-Instruct", 4),
135+
False,
136+
marks=pytest.mark.skip(reason="Skipping due to CI OOM issues")),
137+
pytest.param(
138+
("eagle", "meta-llama/Llama-4-Scout-17B-16E-Instruct",
139+
"morgendave/EAGLE-Llama-4-Scout-17B-16E-Instruct", 4),
140+
True,
141+
marks=pytest.mark.skip(reason="Skipping due to CI OOM issues")),
142+
],
143+
ids=[
144+
"qwen3_eagle3", "llama3_eagle", "llama3_eagle3",
145+
"llama4_eagle", "llama4_eagle_mm"
146+
])
144147
def test_eagle_correctness(
145148
monkeypatch: pytest.MonkeyPatch,
146149
sampling_config: SamplingParams,

0 commit comments

Comments
 (0)