Customized load.
(self, state_dict, use_structured_name=True)
| 1003 | return state_dict_ |
| 1004 | |
| 1005 | def set_state_dict(self, state_dict, use_structured_name=True): |
| 1006 | """Customized load.""" |
| 1007 | |
| 1008 | if self._language_model_key in state_dict: |
| 1009 | state_dict = state_dict[self._language_model_key] |
| 1010 | self.language_model.set_state_dict(state_dict, use_structured_name=use_structured_name) |
nothing calls this directly
no test coverage detected