From d828bc0c2d6d412a333ac191827e31c67a6eba5b Mon Sep 17 00:00:00 2001
From: Goekdeniz-Guelmez <Gulmezdeniz1999@gmail.com>
Date: Tue, 4 Mar 2025 22:42:49 +0100
Subject: [PATCH] remove sanitize method

---
 llms/mlx_lm/models/olmoe.py | 5 -----
 1 file changed, 5 deletions(-)

diff --git a/llms/mlx_lm/models/olmoe.py b/llms/mlx_lm/models/olmoe.py
index 73d9b596..d5d9b425 100644
--- a/llms/mlx_lm/models/olmoe.py
+++ b/llms/mlx_lm/models/olmoe.py
@@ -198,11 +198,6 @@ class Model(nn.Module):
         else:
             out = self.lm_head(out)
         return out
-    
-    def sanitize(self, weights):
-        return {
-            k: v for k, v in weights.items() if "self_attn.rotary_emb.inv_freq" not in k
-        }
 
     @property
     def layers(self):