Commit 04d33670 authored by zhuwenwen's avatar zhuwenwen
Browse files

remove unused code

parent 510401e2
...@@ -576,9 +576,6 @@ class LlamaForCausalLM(nn.Module, SupportsLoRA): ...@@ -576,9 +576,6 @@ class LlamaForCausalLM(nn.Module, SupportsLoRA):
"mlp.down_proj.weight", "mlp.down_proj.weight",
"lm_head.weight" "lm_head.weight"
] ]
if self.use_lm_nn:
lay_key_words.append("lm_head.weight")
combined_words = "|".join(lay_key_words) combined_words = "|".join(lay_key_words)
......
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