MCPcopy Create free account
hub / github.com/LargeWorldModel/LWM / load_state_dict

Method load_state_dict

lwm/data.py:821–826  ·  view source on GitHub ↗
(self, state_dict)

Source from the content-addressed store, hash-verified

819 )
820
821 def load_state_dict(self, state_dict):
822 if 'config' in state_dict:
823 self.config.update(ConfigDict(state_dict['config']))
824 self._index = state_dict.get('index', self.config.example_index_at_start)
825 self._file_loc = state_dict.get('file_loc', self.config.start_seek_loc)
826 self._total_tokens = state_dict.get('total_tokens', self.config.tokens_count_at_start)
827
828 @property
829 def seq_length(self):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected