zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-19 02:38:09 +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
2 changed files with 5 additions and 1 deletions
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}"
)
self.update(tree_unflatten(weights))
if len(weights) != 0:
self.update(tree_unflatten(weights))
return self
def save_weights(self, file: str):