Unverified Commit 895456c4 authored by Jabasukuriputo Wang's avatar Jabasukuriputo Wang Committed by GitHub
Browse files

change state dict comparison to ref compare

parent f92d6149

modules/sd_disable_initialization.py

+1 −1
Original line number Diff line number Diff line
@@ -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)

CRA Git | Maintained and supported by SUSTech CRA and CCSE