nemo_automodel.components.models#
Convenience model builders for NeMo Automodel.
Currently includes: • build_gpt2_model – returns a GPT-2 causal language model (Flash-Attention-2 by default).
Subpackages#
nemo_automodel.components.models.qwen3_nextnemo_automodel.components.models.gpt_ossnemo_automodel.components.models.biencodernemo_automodel.components.models.qwen2nemo_automodel.components.models.qwen3_vl_moenemo_automodel.components.models.llamanemo_automodel.components.models.commonnemo_automodel.components.models.qwen3_omni_moenemo_automodel.components.models.qwen3_moenemo_automodel.components.models.glm4_moenemo_automodel.components.models.deepseek_v3
Submodules#
Package Contents#
Data#
API#
- nemo_automodel.components.models.__all__#
[‘build_gpt2_model’]