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

loading empty list is ok when strict = false (#1834)

Browse Source
This commit is contained in:
Awni Hannun 2025-02-05 16:19:27 -08:00 committed by GitHub
parent fe5987b81d
commit ca305afdbe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -192,7 +192,8 @@ class Module(dict):
f"shape {v_new.shape} for parameter {k}" f"shape {v_new.shape} for parameter {k}"
) )
self.update(tree_unflatten(weights)) if len(weights) != 0:
self.update(tree_unflatten(weights))
return self return self
def save_weights(self, file: str): def save_weights(self, file: str):

3
python/tests/test_nn.py
View File

@ -167,6 +167,9 @@ class TestBase(mlx_tests.MLXTestCase):
] ]
) )
# Empty weights is ok if strict is false
m.load_weights([], strict=False)
def test_module_state(self): def test_module_state(self):
m = nn.Linear(10, 1) m = nn.Linear(10, 1)
m.state["hello"] = "world" m.state["hello"] = "world"