fix module attribute set, reset, set (#1403)

2025-09-20 03:48:15 +08:00 · 2024-09-11 16:30:42 -07:00
parent 02efb310ca
commit 8b30acd7eb
2 changed files with 7 additions and 0 deletions
--- a/python/mlx/nn/layers/base.py
+++ b/python/mlx/nn/layers/base.py
@@ -112,6 +112,7 @@ class Module(dict):
            self[key] = val
        else:
            super(Module, self).__setattr__(key, val)
+            self.pop(key, None)

    def load_weights(
        self,