nemo_automodel.components.models.common

Common/shared components for model implementations.

Subpackages

• nemo_automodel.components.models.common.combined_projection
  • nemo_automodel.components.models.common.combined_projection.combined_mlp
  • nemo_automodel.components.models.common.combined_projection.state_dict_adapter
  • nemo_automodel.components.models.common.combined_projection.combined_qkv

Package Contents

Data

__all__

API

nemo_automodel.components.models.common.__all__

[‘CombinedQKVAttentionMixin’, ‘CombinedGateUpMLP’, ‘CombinedProjectionStateDictAdapter’]