cerebras.modelzoo.tools.checkpoint_converters.helper.maybe_tie_lm_head

cerebras.modelzoo.tools.checkpoint_converters.helper.maybe_tie_lm_head(old_key: str, new_key: str, old_state_dict: collections.OrderedDict, new_state_dict: collections.OrderedDict, from_index: int, action_fn_args: Optional[dict] = None) → None

previous

cerebras.modelzoo.tools.checkpoint_converters.helper.convert_use_rms_layer_norm_helper

next

cerebras.modelzoo.tools.checkpoint_converters.helper.tie_none_weights