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

chore(mlx-lm): add missing model_type for starcoder2 (#522)

Browse Source
This commit is contained in:
Anchen
2024-03-04 01:07:45 +11:00
committed by GitHub
parent 3655bfc3bd
commit 1e3daea3bb
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -167,6 +167,7 @@ class Starcoder2Model(nn.Module):
class Model(nn.Module): class Model(nn.Module):
def __init__(self, args: ModelArgs): def __init__(self, args: ModelArgs):
super().__init__() super().__init__()
self.model_type = args.model_type
self.model = Starcoder2Model(args) self.model = Starcoder2Model(args)
# This is for 15B starcoder2 since it doesn't tie word embeddings # This is for 15B starcoder2 since it doesn't tie word embeddings
if not args.tie_word_embeddings: if not args.tie_word_embeddings: