Revert "fix some typos" (#6244)
This commit is contained in:
@@ -393,7 +393,7 @@ class CohereForCausalLM(nn.Module):
|
||||
weight_loader(param, loaded_weight, shard_id)
|
||||
break
|
||||
else:
|
||||
# lm_head is not used in vLLM as it is tied with embed_token.
|
||||
# lm_head is not used in vllm as it is tied with embed_token.
|
||||
# To prevent errors, skip loading lm_head.weight.
|
||||
if "lm_head.weight" in name:
|
||||
continue
|
||||
|
||||
Reference in New Issue
Block a user