Skip to content

Commit 4ea16c9

Browse files
committed
fix test_initialization
Signed-off-by: wwl2755 <wangwenlong2755@gmail.com>
1 parent 6857d9a commit 4ea16c9

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

vllm/config/__init__.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2165,8 +2165,13 @@ def __post_init__(self):
21652165
# Automatically detect the method
21662166
if self.method in ('eagle', 'eagle3'):
21672167
pass
2168+
# examples:
2169+
# yuhuili/EAGLE-LLaMA3-Instruct-8B
2170+
# yuhuili/EAGLE3-LLaMA3.1-Instruct-8B
2171+
# AngelSlim/Qwen3-8B_eagle3
21682172
elif "eagle-" in self.draft_model_config.model.lower() or \
2169-
"eagle3-" in self.draft_model_config.model.lower():
2173+
"eagle3-" in self.draft_model_config.model.lower() or \
2174+
self.draft_model_config.model.endswith("_eagle3"):
21702175
self.method = "eagle"
21712176
elif self.draft_model_config.hf_config.model_type == "medusa":
21722177
self.method = "medusa"

0 commit comments

Comments
 (0)