zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-21 20:46:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added support for python copy

Browse Source
This commit is contained in:
Manish Aradwad 2024-01-01 17:20:03 +00:00
parent e09bf35b28
commit 1b5e58cec3
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
python/mlx/nn/layers/base.py
View File

@ -439,3 +439,11 @@ class Module(dict):
def eval(self): def eval(self):
self.train(False) self.train(False)
def __getstate__(self):
state = self.__dict__.copy()
state.pop("_compiled_call_impl", None)
return state
def __setstate__(self, state):
self.__dict__.update(state)