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

precommit changes

Browse Source
This commit is contained in:
Manish Aradwad 2024-01-01 17:23:10 +00:00
parent 1b5e58cec3
commit 3301769e0b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -444,6 +444,6 @@ class Module(dict):
state = self.__dict__.copy() state = self.__dict__.copy()
state.pop("_compiled_call_impl", None) state.pop("_compiled_call_impl", None)
return state return state
def __setstate__(self, state): def __setstate__(self, state):
self.__dict__.update(state) self.__dict__.update(state)