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

chore: inform

Browse Source
This commit is contained in:
FL33TW00D 2025-06-10 13:22:52 +01:00
parent 9ce77798b1
commit 43708f3d97
No known key found for this signature in database
GPG Key ID: 640D8032800065F0
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -174,11 +174,13 @@ class Module(dict):
new_weights = dict(weights) new_weights = dict(weights)
curr_weights = dict(tree_flatten(self.parameters())) curr_weights = dict(tree_flatten(self.parameters()))
if extras := (new_weights.keys() - curr_weights.keys()): if extras := (new_weights.keys() - curr_weights.keys()):
extras = " ".join(extras) num_extra = len(extras)
raise ValueError(f"Received parameters not in model: {extras}.") extras = ",\n".join(sorted(extras))
raise ValueError(f"Received {num_extra} parameters not in model: \n{extras}.")
if missing := (curr_weights.keys() - new_weights.keys()): if missing := (curr_weights.keys() - new_weights.keys()):
missing = " ".join(missing) num_missing = len(missing)
raise ValueError(f"Missing parameters: {missing}.") missing = ",\n".join(sorted(missing))
raise ValueError(f"Missing {num_missing} parameters: \n{missing}.")
for k, v in curr_weights.items(): for k, v in curr_weights.items():
v_new = new_weights[k] v_new = new_weights[k]
if not isinstance(v_new, mx.array): if not isinstance(v_new, mx.array):