diff --git a/vllm/model_executor/models/transformers.py b/vllm/model_executor/models/transformers.py index bdc390689104..70daadf91379 100644 --- a/vllm/model_executor/models/transformers.py +++ b/vllm/model_executor/models/transformers.py @@ -345,9 +345,11 @@ class TransformersModel(nn.Module): params_dict = dict(self.named_parameters()) loaded_params = set[str]() for name, loaded_weight in weights: - # Necessary for some models which use remote code - if not name.startswith(prefix := self.model.base_model_prefix): - name = maybe_prefix(prefix, name) + # Use "model" instead of base_model_prefix because + # the base model attribute in vLLM is always `model` + if not name.startswith(prefix := "model."): + name = prefix + name + if is_pp_missing_parameter(name, self): continue if name in params_dict: