diff options
| author | Jabasukuriputo Wang <wfjsw@users.noreply.github.com> | 2023-12-05 18:00:48 -0600 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-05 18:00:48 -0600 |
| commit | 895456c4a2e87f5fe3ee23b4482e68fce317a1ca (patch) | |
| tree | 4c0d0a5c72954a0bb17a41b06fe0b4f25891e223 | |
| parent | f92d61497a426a19818625c3ccdaae9beeb82b31 (diff) | |
change state dict comparison to ref compare
| -rw-r--r-- | modules/sd_disable_initialization.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_disable_initialization.py b/modules/sd_disable_initialization.py index 8863107a..273a7edd 100644 --- 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)
|