Update llms/mlx_lm/models/gemma3_text.py

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
This commit is contained in:
Prince Canuma 2025-03-12 14:59:45 +01:00 committed by GitHub
parent 822546dc91
commit dc090f26df
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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 = []