cerebras.modelzoo.tools.checkpoint_converters.helper.maybe_tie_lm_head

cerebras.modelzoo.tools.checkpoint_converters.helper.maybe_tie_lm_head(old_key, new_key, old_state_dict, new_state_dict, from_index, action_fn_args=None)

previous

cerebras.modelzoo.tools.checkpoint_converters.helper.convert_use_rms_layer_norm_helper

next

cerebras.modelzoo.tools.checkpoint_converters.helper.tie_none_weights