cerebras.modelzoo.tools.checkpoint_converters.helper.tie_none_weights

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

Ties weights stored at old_key & new_key

previous

cerebras.modelzoo.tools.checkpoint_converters.helper.maybe_tie_lm_head

next

cerebras.modelzoo.tools.checkpoint_converters.helper.transpose_key_if_2D