Merge pull request #14216 from wfjsw/state-dict-ref-comparison

change state dict comparison to ref compare
2023-12-14 10:03:14 +03:00 · 2023-12-14 10:03:14 +03:00 · eb52c803b8
parent f8871dedcf
commit eb52c803b8
1 changed files with 1 additions and 1 deletions
--- a/modules/sd_disable_initialization.py
+++ b/modules/sd_disable_initialization.py
@ -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)