bridge.models.falcon_h1.falconh1_bridge#

Module Contents#

Classes#

FalconH1Bridge

Megatron Bridge for FalconH1 Causal LM.

Data#

logger

API#

bridge.models.falcon_h1.falconh1_bridge.logger#

‘getLogger(…)’

class bridge.models.falcon_h1.falconh1_bridge.FalconH1Bridge#

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

Megatron Bridge for FalconH1 Causal LM.

Handles conversion between HuggingFace FalconH1ForCausalLM and Megatron FalconH1Model formats, including weight mappings and configuration translation.

.. rubric:: Example

from megatron.bridge import AutoBridge bridge = AutoBridge.from_hf_pretrained(“tiiuae/Falcon-H1-7B-Instruct”, trust_remote_code=True) provider = bridge.to_megatron_provider()

CONFIG_MAPPING#

None

provider_bridge(
hf_pretrained: megatron.bridge.models.hf_pretrained.causal_lm.PreTrainedCausalLM,
) → megatron.bridge.models.falcon_h1.falconh1_provider.FalconH1ModelProvider#

Convert HuggingFace Falcon H1 config to FalconH1ModelProvider.

classmethod megatron_to_hf_config(
provider: megatron.bridge.models.falcon_h1.falconh1_provider.FalconH1ModelProvider,
) → dict#

Convert FalconH1ModelProvider to HuggingFace config fields.

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

Define parameter mappings between Megatron and HuggingFace formats.