diff --git a/vllm/model_executor/models/phi3_small.py b/vllm/model_executor/models/phi3_small.py index a8b7e9b2a5958..873e9d37771da 100644 --- a/vllm/model_executor/models/phi3_small.py +++ b/vllm/model_executor/models/phi3_small.py @@ -476,6 +476,8 @@ class Phi3SmallForCausalLM(nn.Module, SupportsPP): continue if is_pp_missing_parameter(name, self): continue + if "lm_head.weight" in name and self.config.tie_word_embeddings: + continue param = params_dict[name] weight_loader = getattr(param, "weight_loader", default_weight_loader)