bridge.models.nemotronh.nemotron_h_bridge#

Module Contents#

Classes#

NemotronHBridge

Megatron Bridge for Nemotron-H Causal LM.

Data#

logger

API#

bridge.models.nemotronh.nemotron_h_bridge.logger#

‘getLogger(…)’

class bridge.models.nemotronh.nemotron_h_bridge.NemotronHBridge#

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

Megatron Bridge for Nemotron-H Causal LM.

This bridge handles the conversion between HuggingFace NemotronHForCausalLM and Megatron-Core MambaModel formats, including weight mappings and configuration translation.

.. rubric:: Example

from megatron.bridge import AutoBridge bridge = AutoBridge.from_hf_pretrained(“nvidia/Nemotron-H-8B-Base-8K”, trust_remote_code=True) provider = bridge.to_megatron_provider()

provider_bridge(
hf_pretrained: megatron.bridge.models.hf_pretrained.causal_lm.PreTrainedCausalLM,
) → megatron.bridge.models.nemotronh.nemotron_h_provider.NemotronHModelProvider#
mapping_registry() → megatron.bridge.models.conversion.mapping_registry.MegatronMappingRegistry#