MCPcopy Create free account
hub / github.com/deepspeedai/DeepSpeed / get_module

Method get_module

deepspeed/runtime/state_dict_factory.py:149–155  ·  view source on GitHub ↗
(self, sd)

Source from the content-addressed store, hash-verified

147 return 'model'
148
149 def get_module(self, sd):
150 if self.module_key is None:
151 return sd
152 elif self.module_key == AUTO_MODULE_KEY:
153 return sd[self._choose_module_key(sd)]
154 else:
155 return sd[self.module_key]
156
157 def set_module(self, sd, module):
158 if self.module_key is None:

Callers 4

loadMethod · 0.95
merge_state_dictMethod · 0.80
split_state_dictMethod · 0.80
sanity_checkMethod · 0.80

Calls 1

_choose_module_keyMethod · 0.95

Tested by

no test coverage detected