Unverified Commit 6469038b authored by Michael Greenbaum's avatar Michael Greenbaum Committed by GitHub
Browse files

[Bugfix] Fix loading of fine-tuned models based on Phi-3-Small (#12689)


Signed-off-by: default avatarMichael Greenbaum <mgreenbaum@microsoft.com>
Co-authored-by: default avatarMichael Greenbaum <mgreenbaum@microsoft.com>
parent 815079de
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment