From 799dfde8a38e34a8da2e6858dbd6531f792a3d12 Mon Sep 17 00:00:00 2001
From: Awni Hannun <awni@apple.com>
Date: Mon, 16 Dec 2024 07:55:06 -0800
Subject: [PATCH] nits

---
 llms/mlx_lm/models/cohere2.py | 8 --------
 1 file changed, 8 deletions(-)

diff --git a/llms/mlx_lm/models/cohere2.py b/llms/mlx_lm/models/cohere2.py
index 392c7633..fcb4061b 100644
--- a/llms/mlx_lm/models/cohere2.py
+++ b/llms/mlx_lm/models/cohere2.py
@@ -205,11 +205,3 @@ class Model(nn.Module):
     @property
     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