zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-24 09:21:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixes Typo in Starcoder2 (#740)

Browse Source
This commit is contained in:
Thomas Lazarus 2024-04-29 15:14:45 -05:00 committed by GitHub
parent 510d2bde49
commit 5513c4e57d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llms/mlx_lm/models/starcoder2.py
View File

@ -150,7 +150,7 @@ class Model(nn.Module):
self.model_type = args.model_type
self.model = Starcoder2Model(args)
if not args.tie_word_embeddings:
sself.lm_head = nn.Linear(args.hidden_size, args.vocab_size, bias=False)
self.lm_head = nn.Linear(args.hidden_size, args.vocab_size, bias=False)
def __call__(
self,