mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-08-10 02:58:06 +08:00
Update llms/mlx_lm/models/gemma3_text.py
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
This commit is contained in:
parent
822546dc91
commit
dc090f26df
@ -226,13 +226,6 @@ class Model(nn.Module):
|
||||
def layers(self):
|
||||
return self.model.layers
|
||||
|
||||
@property
|
||||
def head_dim(self):
|
||||
return self.args.head_dim
|
||||
|
||||
@property
|
||||
def n_kv_heads(self):
|
||||
return self.args.num_key_value_heads
|
||||
|
||||
def make_cache(self):
|
||||
caches = []
|
||||
|
Loading…
Reference in New Issue
Block a user