@@ -123,24 +123,27 @@ def test_ngram_correctness(
123
123
cleanup_dist_env_and_memory ()
124
124
125
125
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
+ ])
144
147
def test_eagle_correctness (
145
148
monkeypatch : pytest .MonkeyPatch ,
146
149
sampling_config : SamplingParams ,
0 commit comments