@@ -125,24 +125,27 @@ def test_ngram_correctness(
125125 cleanup_dist_env_and_memory ()
126126
127127
128- @pytest .mark .parametrize (
129- ["model_setup" , "mm_enabled" ], [
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- ],
145- ids = ["llama3_eagle" , "llama3_eagle3" , "llama4_eagle" , "llama4_eagle_mm" ])
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+ ],
145+ ids = [
146+ "qwen3_eagle3" , "llama3_eagle" , "llama3_eagle3" ,
147+ "llama4_eagle" , "llama4_eagle_mm"
148+ ])
146149@pytest .mark .parametrize ("attn_backend" ,
147150 get_attn_backend_list_based_on_platform ())
148151def test_eagle_correctness (
0 commit comments