mcore_hf_export_map

Custom mapping from Megatron Core models to their Hugging Face counter part.

Classes

CustomModuleMapping

A custom module mapping from Megatron Core to its HF counter part.

class CustomModuleMapping

Bases: object

A custom module mapping from Megatron Core to its HF counter part.

__init__(func_name='', target_name_or_prefix='', func_kwargs={})

Create a custom module mapping.

Parameters:

• func_name (str) –

• target_name_or_prefix (str) –

• func_kwargs (Dict[str, Any]) –