diff --git a/vllm/config.py b/vllm/config.py index f86c3272a0ad..336a31502ce4 100644 --- a/vllm/config.py +++ b/vllm/config.py @@ -2335,7 +2335,9 @@ def __post_init__(self): ) # Automatically detect the method - if "eagle-" in self.draft_model_config.model.lower(): + if self.method == 'eagle': + pass + elif "eagle-" in self.draft_model_config.model.lower(): self.method = "eagle" elif self.draft_model_config.hf_config.model_type == "medusa": self.method = "medusa"