From dda3aca47f0956691c4de712c23f01125e4d70e4 Mon Sep 17 00:00:00 2001 From: Orestis Zambounis <23146389+orestis-z@users.noreply.github.com> Date: Thu, 25 Jun 2026 02:51:33 +0200 Subject: [PATCH] [Speculative Decoding] Propagate norm_output and fc_norm config for Eagle3 speculators (#46488) Signed-off-by: Orestis Zambounis Co-authored-by: Claude Opus 4.6 Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com> --- vllm/transformers_utils/configs/speculators/algos.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/vllm/transformers_utils/configs/speculators/algos.py b/vllm/transformers_utils/configs/speculators/algos.py index 0dc3ccce08..f1dfc8878f 100644 --- a/vllm/transformers_utils/configs/speculators/algos.py +++ b/vllm/transformers_utils/configs/speculators/algos.py @@ -36,6 +36,8 @@ def update_eagle3(config_dict: dict, pre_trained_config: dict) -> None: "norm_before_residual", True ) pre_trained_config["norm_before_fc"] = config_dict.get("norm_before_fc", False) + pre_trained_config["fc_norm"] = config_dict.get("fc_norm", False) + pre_trained_config["norm_output"] = config_dict.get("norm_output", False) eagle3_arch_map = { "qwen3": "Eagle3Qwen3ForCausalLM", "llama": "Eagle3LlamaForCausalLM",