* fix typos

* ci

---------

Co-authored-by: YuBaoku <49938469+EmmonsCurse@users.noreply.github.com>
This commit is contained in:
co63oc
2025-09-12 11:04:38 +08:00
committed by GitHub
parent 82dab8a91a
commit 8466219ec8
14 changed files with 22 additions and 22 deletions

View File

@@ -56,7 +56,7 @@ class ParallelLMHead(nn.Layer):
embedding_dim (int): size of hidden state.
prefix (str): The name of current layer. Defaults to "".
with_bias (bool): whether to have bias. Default: False.
dtype (str): The dtype of weight. Defalut: None.
dtype (str): The dtype of weight. Default: None.
"""
super(ParallelLMHead, self).__init__()
self.weight_key: str = prefix + ".weight"