Skip to content

Commit 6857d9a

Browse files
committed
enable test in CI
Signed-off-by: wwl2755 <wangwenlong2755@gmail.com>
1 parent 0bc7ecd commit 6857d9a

File tree

1 file changed

+24
-30
lines changed

1 file changed

+24
-30
lines changed

tests/v1/e2e/test_spec_decode.py

Lines changed: 24 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -125,36 +125,30 @@ def test_ngram_correctness(
125125
cleanup_dist_env_and_memory()
126126

127127

128-
@pytest.mark.parametrize(
129-
["model_setup", "mm_enabled"],
130-
[
131-
(("eagle3", "Qwen/Qwen3-8B", "AngelSlim/Qwen3-8B_eagle3", 1), False),
132-
(("eagle", "meta-llama/Llama-3.1-8B-Instruct",
133-
"yuhuili/EAGLE-LLaMA3.1-Instruct-8B", 1), False),
134-
(("eagle3", "meta-llama/Llama-3.1-8B-Instruct",
135-
"yuhuili/EAGLE3-LLaMA3.1-Instruct-8B", 1), False),
136-
pytest.param(
137-
("eagle", "meta-llama/Llama-4-Scout-17B-16E-Instruct",
138-
"morgendave/EAGLE-Llama-4-Scout-17B-16E-Instruct", 4),
139-
False,
140-
marks=pytest.mark.skip(reason="Skipping due to CI OOM issues")),
141-
pytest.param(
142-
("eagle", "meta-llama/Llama-4-Scout-17B-16E-Instruct",
143-
"morgendave/EAGLE-Llama-4-Scout-17B-16E-Instruct", 4),
144-
True,
145-
marks=pytest.mark.skip(reason="Skipping due to CI OOM issues")),
146-
(("eagle", "eagle618/deepseek-v3-random",
147-
"eagle618/eagle-deepseek-v3-random", 1), False),
148-
],
149-
ids=[
150-
# TODO: Re-enable this once tests/models/test_initialization.py is fixed, see PR #22333 #22611 # noqa: E501
151-
"qwen3_eagle3",
152-
"llama3_eagle",
153-
"llama3_eagle3",
154-
"llama4_eagle",
155-
"llama4_eagle_mm",
156-
"deepseek_eagle"
157-
])
128+
@pytest.mark.parametrize(["model_setup", "mm_enabled"], [
129+
(("eagle3", "Qwen/Qwen3-8B", "AngelSlim/Qwen3-8B_eagle3", 1), False),
130+
(("eagle", "meta-llama/Llama-3.1-8B-Instruct",
131+
"yuhuili/EAGLE-LLaMA3.1-Instruct-8B", 1), False),
132+
(("eagle3", "meta-llama/Llama-3.1-8B-Instruct",
133+
"yuhuili/EAGLE3-LLaMA3.1-Instruct-8B", 1), False),
134+
pytest.param(
135+
("eagle", "meta-llama/Llama-4-Scout-17B-16E-Instruct",
136+
"morgendave/EAGLE-Llama-4-Scout-17B-16E-Instruct", 4),
137+
False,
138+
marks=pytest.mark.skip(reason="Skipping due to CI OOM issues")),
139+
pytest.param(
140+
("eagle", "meta-llama/Llama-4-Scout-17B-16E-Instruct",
141+
"morgendave/EAGLE-Llama-4-Scout-17B-16E-Instruct", 4),
142+
True,
143+
marks=pytest.mark.skip(reason="Skipping due to CI OOM issues")),
144+
(("eagle", "eagle618/deepseek-v3-random",
145+
"eagle618/eagle-deepseek-v3-random", 1), False),
146+
],
147+
ids=[
148+
"qwen3_eagle3", "llama3_eagle", "llama3_eagle3",
149+
"llama4_eagle", "llama4_eagle_mm",
150+
"deepseek_eagle"
151+
])
158152
@pytest.mark.parametrize("attn_backend",
159153
get_attn_backend_list_based_on_platform())
160154
def test_eagle_correctness(

0 commit comments

Comments
 (0)