cerebras.modelzoo.tools.checkpoint_converters.llava.Converter_LLaVA_WithoutModel_HF_CS22#

class cerebras.modelzoo.tools.checkpoint_converters.llava.Converter_LLaVA_WithoutModel_HF_CS22[source]#

Bases: cerebras.modelzoo.tools.checkpoint_converters.base_converter.BaseCheckpointConverter_UnpackedHF_PackedCS

Methods

`architectures`
`attempt_mup_to_sp`
`component_names`
`convert`
`convert_all_keys`
`convert_helper`
`convert_key`	Attempts to convert the old key by matching against the list of conversion rules.
`converter_note`
`converters`
`extract_model_dict`
`file_formats`
`formats`
`get_config_converter_class`
`get_converter_indices`
`get_mup_converter`	Allows models to override the default muP converters with their own.
`init_output_checkpoint`
`load`
`match_indices`
`post_checkpoint_convert`
`post_model_convert`
`pre_checkpoint_convert`
`pre_model_convert`	Hook executes right before model conversion.
`replaceKey`	Copies value that exists at old_state_dict's old_key to new_state_dict's new_key.
`save`
`supports_conversion`
`supports_mup_conversion`

convert_key(old_key, old_state_dict, new_state_dict, from_index, match_start=0, prefix='', action_fn_args=None, debug=False)#

Attempts to convert the old key by matching against the list of conversion rules. The first rule to match is used for conversion (i.e. even if multiple rules would match, the latter ones are never used). Returns True if a conversion occurred.

get_mup_converter()#

Allows models to override the default muP converters with their own.

pre_model_convert(old_state_dict, new_state_dict, configs, converter_indices, drop_unmatched_keys)#

Hook executes right before model conversion.

static replaceKey(old_key, new_key, old_state_dict, new_state_dict, from_index, action_fn_args=None)#

Copies value that exists at old_state_dict’s old_key to new_state_dict’s new_key.

cerebras.modelzoo.tools.checkpoint_converters.llava.Converter_LLaVA_LLaMA_WithoutModel_HF_CS22

cerebras.modelzoo.tools.checkpoint_converters.mistral