fix spec_layer_weight_names

dbcf9f45 · zhangning3 · 84ac1d27 · dbcf9f45
Commit dbcf9f45 authored May 09, 2026 by zhangning3
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

vllm/model_executor/models/step3p5_mtp.py vllm/model_executor/models/step3p5_mtp.py +1 -0

No files found.
--- a/vllm/model_executor/models/step3p5_mtp.py
+++ b/vllm/model_executor/models/step3p5_mtp.py
@@ -306,6 +306,7 @@ class Step3p5MTP(nn.Module):
            "enorm",
            "hnorm",
            "eh_proj",
+            "lm_head",
            "shared_head",
        ]
        spec_layer_weight = False