cerebras.modelzoo.tools.checkpoint_converters.helper

Functions

Build_HF_CS_Converter_WithOptionalModel
convert_use_biasless_layer_norm_helper
convert_use_rms_layer_norm_helper
maybe_tie_lm_head
tie_none_weights	Ties weights stored at old_key & new_key
transpose_key_if_2D

previous

cerebras.modelzoo.tools.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS20

next

cerebras.modelzoo.tools.checkpoint_converters.helper.Build_HF_CS_Converter_WithOptionalModel