forked from EngineX-Cambricon/enginex-mlu370-vllm
debugging
This commit is contained in:
@@ -404,6 +404,12 @@ class LlamaModel(nn.Module):
|
||||
if is_pp_missing_parameter(name, self):
|
||||
continue
|
||||
|
||||
if name not in params_dict:
|
||||
logger.warning(
|
||||
"Skipping weight %s not present in the model",
|
||||
name)
|
||||
continue
|
||||
|
||||
param = params_dict[name]
|
||||
weight_loader = getattr(param, "weight_loader",
|
||||
default_weight_loader)
|
||||
|
||||
Reference in New Issue
Block a user