Merge pull request AUTOMATIC1111#14216 from wfjsw/state-dict-ref-comp…

…arison change state dict comparison to ref compare
ruchej · Sep 30, 2024 · 2271615 · 2271615
1 parent e58f033
commit 2271615
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/sd_disable_initialization.py b/modules/sd_disable_initialization.py
@@ -215,7 +215,7 @@ def load_state_dict(original, module, state_dict, strict=True):
             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)