update to latest version

2026-09-18 16:55:15 +02:00 · 2023-12-16 21:14:48 +02:00
parent 6029303af5
commit 598aef8230
103 changed files with 3507 additions and 1047 deletions
@@ -215,7 +215,7 @@ class LoadStateDictOnMeta(ReplaceHelper):
            would be on the meta device.
            """

-            if state_dict == sd:
+            if state_dict is sd:
                state_dict = {k: v.to(device="meta", dtype=v.dtype) for k, v in state_dict.items()}

            original(module, state_dict, strict=strict)