From 74328a09381c2ad81ff1ff39d13909df52e9ca8b Mon Sep 17 00:00:00 2001
From: Manish Aradwad <manisharadwad@gmail.com>
Date: Tue, 2 Jan 2024 04:16:35 +0000
Subject: [PATCH] removed `_compiled_call_impl` line

---
 python/mlx/nn/layers/base.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/python/mlx/nn/layers/base.py b/python/mlx/nn/layers/base.py
index c22d0adc2..4101111a7 100644
--- a/python/mlx/nn/layers/base.py
+++ b/python/mlx/nn/layers/base.py
@@ -442,7 +442,6 @@ class Module(dict):
 
     def __getstate__(self):
         state = self.__dict__.copy()
-        state.pop("_compiled_call_impl", None)
         return state
 
     def __setstate__(self, state):