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):
|
if is_pp_missing_parameter(name, self):
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
if name not in params_dict:
|
||||||
|
logger.warning(
|
||||||
|
"Skipping weight %s not present in the model",
|
||||||
|
name)
|
||||||
|
continue
|
||||||
|
|
||||||
param = params_dict[name]
|
param = params_dict[name]
|
||||||
weight_loader = getattr(param, "weight_loader",
|
weight_loader = getattr(param, "weight_loader",
|
||||||
default_weight_loader)
|
default_weight_loader)
|
||||||
|
|||||||
Reference in New Issue
Block a user