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,
- 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(…)’