MCPcopy Create free account
hub / github.com/THUDM/GLM / load_state_dict

Method load_state_dict

fp16/fp16.py:81–82  ·  view source on GitHub ↗
(self, state_dict, strict=True)

Source from the content-addressed store, hash-verified

79 return self.module.state_dict(destination, prefix, keep_vars)
80
81 def load_state_dict(self, state_dict, strict=True):
82 return self.module.load_state_dict(state_dict, strict=strict)
83
84
85# TODO: Update overflow check + downscale to use Carl's fused kernel.

Callers 4

load_checkpointFunction · 0.45
load_pretrainedFunction · 0.45
__init__Method · 0.45
load_state_dictMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected