diff --git a/llms/mlx_lm/models/starcoder2.py b/llms/mlx_lm/models/starcoder2.py index 2637a35a..e96db5b8 100644 --- a/llms/mlx_lm/models/starcoder2.py +++ b/llms/mlx_lm/models/starcoder2.py @@ -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,