MCPcopy
hub / github.com/hpcaitech/Open-Sora / load_state_dict

Method load_state_dict

opensora/datasets/sampler.py:326–327  ·  view source on GitHub ↗
(self, state_dict: dict)

Source from the content-addressed store, hash-verified

324 return {"start_index": step}
325
326 def load_state_dict(self, state_dict: dict):
327 self.start_index = state_dict["start_index"] + 1

Callers 12

mainFunction · 0.45
mainFunction · 0.45
load_modelFunction · 0.45
load_checkpointFunction · 0.45
loadFunction · 0.45
load_checkpoint_with_inflationFunction · 0.45
load_from_pretrainedMethod · 0.45
from_pretrainedMethod · 0.45
__init__Method · 0.45
load_from_pretrainedMethod · 0.45
mainFunction · 0.45
test_block_attn_3d_var_kernelFunction · 0.45

Calls

no outgoing calls

Tested by 1

test_block_attn_3d_var_kernelFunction · 0.36