@@ -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 ())
160154def test_eagle_correctness (
0 commit comments