From 5513c4e57dd78bf0d929b49ea2818ee0ac9644b2 Mon Sep 17 00:00:00 2001
From: Thomas Lazarus <46943923+lazarust@users.noreply.github.com>
Date: Mon, 29 Apr 2024 15:14:45 -0500
Subject: [PATCH] Fixes Typo in Starcoder2 (#740)

---
 llms/mlx_lm/models/starcoder2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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,