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)[source]#