`bridge.models.nemotron_vl.nemotron_vl_bridge`#

Module Contents#

Classes#

NemotronVLBridge

Shared Omni mappings and compatibility for historical MoE checkpoint labels.

Functions#

_is_legacy_v2_omni_config

Identify Nano Omni checkpoints serialized with the historical V2 label.

API#

bridge.models.nemotron_vl.nemotron_vl_bridge._is_legacy_v2_omni_config(hf_config) → bool#: Identify Nano Omni checkpoints serialized with the historical V2 label.

class bridge.models.nemotron_vl.nemotron_vl_bridge.NemotronVLBridge#

Bases: megatron.bridge.models.conversion.model_bridge.MegatronModelBridge

Shared Omni mappings and compatibility for historical MoE checkpoint labels.

Dense Nemotron Nano v2 checkpoints are no longer supported.

CONFIG_MAPPING#: None

_canonical_omni_bridge()#: Create the canonical bridge lazily to avoid a module import cycle.

provider_bridge( hf_pretrained: megatron.bridge.models.hf_pretrained.causal_lm.PreTrainedCausalLM, )#

mapping_registry() → megatron.bridge.models.conversion.mapping_registry.MegatronMappingRegistry#

_legacy_mapping_registry() → megatron.bridge.models.conversion.mapping_registry.MegatronMappingRegistry#: Return the historical wrapper-prefixed Nemotron-VL mappings.