MCPcopy Create free account
hub / github.com/YesianRohn/TextSSR / add_state_dict

Function add_state_dict

diffusers/scripts/convert_consistency_decoder.py:1047–1049  ·  view source on GitHub ↗
(prefix, mod)

Source from the content-addressed store, hash-verified

1045
1046
1047def add_state_dict(prefix, mod):
1048 for k, v in mod.state_dict().items():
1049 unet_state_dict[f"{prefix}.{k}"] = v
1050
1051
1052add_state_dict("conv_in", conv_in)

Callers 1

convert_consistency_decoder.pyFile · 0.85

Calls 1

state_dictMethod · 0.45

Tested by

no test coverage detected