nemo_automodel.components._transformers.registry#

Module Contents#

Classes#

Functions#

Data#

API#

nemo_automodel.components._transformers.registry.logger#

‘getLogger(…)’

nemo_automodel.components._transformers.registry.MODELING_PATH#

[‘nemo_automodel.components.models’]

class nemo_automodel.components._transformers.registry._ModelRegistry#
modeling_path: List[str]#

‘field(…)’

model_arch_name_to_cls: Dict[str, Union[Type[torch.nn.Module], str]]#

‘field(…)’

__post_init__()#
property supported_models: Dict[str, Type[torch.nn.Module]]#
get_model_cls_from_model_arch(
model_arch: str,
) Type[torch.nn.Module]#
register_modeling_path(path: str) None#

Add a new modeling path and register models from it.

_mapping_model_arch_name_to_cls(modeling_path: str)#
nemo_automodel.components._transformers.registry.get_registry()#
nemo_automodel.components._transformers.registry.ModelRegistry#

‘get_registry(…)’