bridge.models

Subpackages

• bridge.models.exaone
  • bridge.models.exaone.exaone4
    • bridge.models.exaone.exaone4.exaone4_bridge
    • bridge.models.exaone.exaone4.exaone4_provider
  • bridge.models.exaone.exaone45
    • bridge.models.exaone.exaone45.modelling_exaone45
    • bridge.models.exaone.exaone45.exaone45_provider
    • bridge.models.exaone.exaone45.exaone45_bridge
  • bridge.models.exaone.exaone_moe
    • bridge.models.exaone.exaone_moe.exaone_moe_bridge
    • bridge.models.exaone.exaone_moe.exaone_moe_provider
• bridge.models.muse_spark_open
  • bridge.models.muse_spark_open.modeling_muse_spark_open
    • bridge.models.muse_spark_open.modeling_muse_spark_open.model
    • bridge.models.muse_spark_open.modeling_muse_spark_open.blocks
    • bridge.models.muse_spark_open.modeling_muse_spark_open.muse_spark_open_transformers_patch
  • bridge.models.muse_spark_open.text_only
  • bridge.models.muse_spark_open.muse_spark_open_config
  • bridge.models.muse_spark_open.muse_spark_open_bridge
  • bridge.models.muse_spark_open.muse_spark_open_builder
• bridge.models.glm_moe_dsa
  • bridge.models.glm_moe_dsa.glm5_hybrid
  • bridge.models.glm_moe_dsa.glm5_bridge
  • bridge.models.glm_moe_dsa.glm5_provider
• bridge.models.qwen
  • bridge.models.qwen.qwen3_moe_bridge
  • bridge.models.qwen.qwen2_bridge
  • bridge.models.qwen.qwen3_next_bridge
  • bridge.models.qwen.qwen3_bridge
  • bridge.models.qwen.qwen35_bridge
• bridge.models.glm
  • bridge.models.glm.glm45_bridge
  • bridge.models.glm.glm47_flash_bridge
• bridge.models.glm_vl
  • bridge.models.glm_vl.data
    • bridge.models.glm_vl.data.collate_fn
  • bridge.models.glm_vl.glm_45v_bridge
  • bridge.models.glm_vl.modeling_glm_45v
  • bridge.models.glm_vl.glm_45v_provider
• bridge.models.mistral
  • bridge.models.mistral.mistral_bridge
  • bridge.models.mistral.mistral_provider
• bridge.models.nemotron_vl
  • bridge.models.nemotron_vl.nemotron_vl_bridge
  • bridge.models.nemotron_vl.nemotron_vl_utils
  • bridge.models.nemotron_vl.modeling_nemotron_vl
• bridge.models.decorators
  • bridge.models.decorators.dispatch
  • bridge.models.decorators.torchrun
• bridge.models.gemma_vl
  • bridge.models.gemma_vl.data
    • bridge.models.gemma_vl.data.collate_fn
  • bridge.models.gemma_vl.gemma3_vl_bridge
  • bridge.models.gemma_vl.modeling_gemma4_vl
  • bridge.models.gemma_vl.gemma3_vl_provider
  • bridge.models.gemma_vl.gemma4_vl_provider
  • bridge.models.gemma_vl.modeling_gemma3_vl
  • bridge.models.gemma_vl.gemma4_vl_bridge
• bridge.models.stepfun
  • bridge.models.stepfun.modelling_step37
    • bridge.models.stepfun.modelling_step37.utils
    • bridge.models.stepfun.modelling_step37.vision_model
    • bridge.models.stepfun.modelling_step37.transformer_block
    • bridge.models.stepfun.modelling_step37.attention
    • bridge.models.stepfun.modelling_step37.model
    • bridge.models.stepfun.modelling_step37.rope
    • bridge.models.stepfun.modelling_step37.image_insert_embedding
    • bridge.models.stepfun.modelling_step37.transformer_config
    • bridge.models.stepfun.modelling_step37.text_model
  • bridge.models.stepfun.data
    • bridge.models.stepfun.data.flickr8k
  • bridge.models.stepfun.step37_provider
  • bridge.models.stepfun.configuration_step35
  • bridge.models.stepfun.step37_flickr8k_step
  • bridge.models.stepfun.step37_bridge
  • bridge.models.stepfun.step35_provider
  • bridge.models.stepfun.step35_bridge
  • bridge.models.stepfun.configuration_step37
• bridge.models.megatron_mimo
  • bridge.models.megatron_mimo.conversion
    • bridge.models.megatron_mimo.conversion.mimo_model_io
    • bridge.models.megatron_mimo.conversion.orchestrator
  • bridge.models.megatron_mimo.megatron_mimo_ddp
  • bridge.models.megatron_mimo.llava_provider
  • bridge.models.megatron_mimo.megatron_mimo_provider
  • bridge.models.megatron_mimo.megatron_mimo_builder
  • bridge.models.megatron_mimo.build_model
  • bridge.models.megatron_mimo.megatron_mimo_config
• bridge.models.qwen3_asr
  • bridge.models.qwen3_asr.hf_qwen3_asr
    • bridge.models.qwen3_asr.hf_qwen3_asr.configuration_qwen3_asr
    • bridge.models.qwen3_asr.hf_qwen3_asr.processing_qwen3_asr
    • bridge.models.qwen3_asr.hf_qwen3_asr.modeling_qwen3_asr
  • bridge.models.qwen3_asr.modeling_qwen3_asr
    • bridge.models.qwen3_asr.modeling_qwen3_asr.model
    • bridge.models.qwen3_asr.modeling_qwen3_asr.rope
    • bridge.models.qwen3_asr.modeling_qwen3_asr.transformer_config
    • bridge.models.qwen3_asr.modeling_qwen3_asr.thinker_model
  • bridge.models.qwen3_asr.qwen3_asr_bridge
  • bridge.models.qwen3_asr.qwen3_asr_provider
• bridge.models.gpt
  • bridge.models.gpt.model_config
  • bridge.models.gpt.gpt_builder
• bridge.models.bagel
  • bridge.models.bagel.data
    • bridge.models.bagel.data.dataset
    • bridge.models.bagel.data.external
    • bridge.models.bagel.data.packing
    • bridge.models.bagel.data.energon
    • bridge.models.bagel.data.batch
    • bridge.models.bagel.data.order
  • bridge.models.bagel.modeling
  • bridge.models.bagel.provider
  • bridge.models.bagel.checkpoint
  • bridge.models.bagel.bagel_step
  • bridge.models.bagel.conversion
  • bridge.models.bagel.dependencies
  • bridge.models.bagel.diffusion
• bridge.models.ernie
  • bridge.models.ernie.ernie_45_bridge
• bridge.models.ernie_vl
  • bridge.models.ernie_vl.modeling_ernie45_vl
    • bridge.models.ernie_vl.modeling_ernie45_vl.ernie_decoder_layer_spec
    • bridge.models.ernie_vl.modeling_ernie45_vl.vision_attention
    • bridge.models.ernie_vl.modeling_ernie45_vl.vision_layer_spec
    • bridge.models.ernie_vl.modeling_ernie45_vl.vision_model
    • bridge.models.ernie_vl.modeling_ernie45_vl.ernie_moe_layer
    • bridge.models.ernie_vl.modeling_ernie45_vl.vision_transformer_config
    • bridge.models.ernie_vl.modeling_ernie45_vl.model
  • bridge.models.ernie_vl.ernie45_vl_bridge
  • bridge.models.ernie_vl.ernie45_vl_provider
• bridge.models.kimi_vl
  • bridge.models.kimi_vl.data
    • bridge.models.kimi_vl.data.collate_fn
  • bridge.models.kimi_vl.modeling_kimi_k25_vl
  • bridge.models.kimi_vl.utils
  • bridge.models.kimi_vl.kimi_k25_vl_bridge
  • bridge.models.kimi_vl.kimi_k25_vl_provider
• bridge.models.mimo_v2_flash
  • bridge.models.mimo_v2_flash.mimo_v2_flash_provider
  • bridge.models.mimo_v2_flash.modeling_mimo_v2_flash
  • bridge.models.mimo_v2_flash.mimo_v2_flash_bridge
• bridge.models.kimi
  • bridge.models.kimi.kimi_k3_pipeline
  • bridge.models.kimi.kimi_bridge
  • bridge.models.kimi.kimi_k3_spec
  • bridge.models.kimi.kimi_k3_ops
  • bridge.models.kimi.kimi_k3_bridge
  • bridge.models.kimi.kimi_k3_layers
  • bridge.models.kimi.kimi_k3_provider
• bridge.models.gpt_oss
  • bridge.models.gpt_oss.gpt_oss_bridge
• bridge.models.gemma
  • bridge.models.gemma.gemma4_bridge
  • bridge.models.gemma.gemma4_provider
  • bridge.models.gemma.gemma3_provider
  • bridge.models.gemma.modules
  • bridge.models.gemma.modeling_gemma4
  • bridge.models.gemma.gemma3_bridge
• bridge.models.qwen_audio
  • bridge.models.qwen_audio.data
    • bridge.models.qwen_audio.data.collate_fn
  • bridge.models.qwen_audio.qwen2_audio_provider
  • bridge.models.qwen_audio.modeling_qwen2_audio
  • bridge.models.qwen_audio.qwen2_audio_bridge
• bridge.models.mimo
  • bridge.models.mimo.mimo_bridge
• bridge.models.hy_v3
  • bridge.models.hy_v3.hy_v3_bridge
• bridge.models.beam
  • bridge.models.beam.beam_provider
  • bridge.models.beam.beam_builder
  • bridge.models.beam.beam_mappings
  • bridge.models.beam.beam_config
  • bridge.models.beam.modeling_beam
  • bridge.models.beam.beam_bridge
• bridge.models.falcon_h1
  • bridge.models.falcon_h1.falconh1_provider
  • bridge.models.falcon_h1.falconh1_bridge
• bridge.models.hf_pretrained
  • bridge.models.hf_pretrained.safe_config_loader
  • bridge.models.hf_pretrained.utils
  • bridge.models.hf_pretrained.token_classification
  • bridge.models.hf_pretrained.causal_lm
  • bridge.models.hf_pretrained.state
  • bridge.models.hf_pretrained.masked_lm
  • bridge.models.hf_pretrained.base
• bridge.models.hybrid
  • bridge.models.hybrid.hybrid_provider
  • bridge.models.hybrid.hybrid_builder
• bridge.models.diffusion_gemma
  • bridge.models.diffusion_gemma.modeling_diffusion_gemma
  • bridge.models.diffusion_gemma.diffusion_gemma_step
  • bridge.models.diffusion_gemma.diffusion_gemma_bridge
  • bridge.models.diffusion_gemma.diffusion_gemma_provider
  • bridge.models.diffusion_gemma.loss
  • bridge.models.diffusion_gemma.data
• bridge.models.sarvam
  • bridge.models.sarvam.sarvam_provider
  • bridge.models.sarvam.sarvam_moe_bridge
  • bridge.models.sarvam.sarvam_mla_bridge
  • bridge.models.sarvam.common
• bridge.models.qwen_omni
  • bridge.models.qwen_omni.modeling_qwen25_omni
    • bridge.models.qwen_omni.modeling_qwen25_omni.model
    • bridge.models.qwen_omni.modeling_qwen25_omni.rope
    • bridge.models.qwen_omni.modeling_qwen25_omni.transformer_config
    • bridge.models.qwen_omni.modeling_qwen25_omni.thinker_model
  • bridge.models.qwen_omni.data
    • bridge.models.qwen_omni.data.collate_fn
  • bridge.models.qwen_omni.modeling_qwen3_omni
    • bridge.models.qwen_omni.modeling_qwen3_omni.model
    • bridge.models.qwen_omni.modeling_qwen3_omni.rope
    • bridge.models.qwen_omni.modeling_qwen3_omni.transformer_config
    • bridge.models.qwen_omni.modeling_qwen3_omni.thinker_model
  • bridge.models.qwen_omni.qwen25_omni_provider
  • bridge.models.qwen_omni.qwen25_omni_bridge
  • bridge.models.qwen_omni.qwen3_omni_provider
  • bridge.models.qwen_omni.qwen3_omni_bridge
  • bridge.models.qwen_omni.qwen3_omni_step
• bridge.models.nemotronh
  • bridge.models.nemotronh.nemotron_h_bridge
• bridge.models.bert
  • bridge.models.bert.bert_provider
  • bridge.models.bert.bert_bridge
• bridge.models.minimax_m3
  • bridge.models.minimax_m3.transformers_compat
  • bridge.models.minimax_m3.modeling_minimax_m3_vl
  • bridge.models.minimax_m3.minimax_m3_bridge
• bridge.models.common
  • bridge.models.common.te_layers
  • bridge.models.common.unimodal
  • bridge.models.common.heads
  • bridge.models.common.base
• bridge.models.qwen_vl
  • bridge.models.qwen_vl.data
    • bridge.models.qwen_vl.data.energon
    • bridge.models.qwen_vl.data.collate_fn
  • bridge.models.qwen_vl.modelling_qwen3_vl
    • bridge.models.qwen_vl.modelling_qwen3_vl.utils
    • bridge.models.qwen_vl.modelling_qwen3_vl.vision_model
    • bridge.models.qwen_vl.modelling_qwen3_vl.token_classification
    • bridge.models.qwen_vl.modelling_qwen3_vl.transformer_block
    • bridge.models.qwen_vl.modelling_qwen3_vl.attention
    • bridge.models.qwen_vl.modelling_qwen3_vl.model
    • bridge.models.qwen_vl.modelling_qwen3_vl.rope
    • bridge.models.qwen_vl.modelling_qwen3_vl.transformer_config
    • bridge.models.qwen_vl.modelling_qwen3_vl.text_model
  • bridge.models.qwen_vl.qwen3_vl_step
  • bridge.models.qwen_vl.modeling_qwen25_vl
  • bridge.models.qwen_vl.qwen25_vl_provider
  • bridge.models.qwen_vl.qwen35_vl_provider
  • bridge.models.qwen_vl.qwen3_vl_bridge
  • bridge.models.qwen_vl.qwen3_vl_provider
  • bridge.models.qwen_vl.qwen35_vl_bridge
  • bridge.models.qwen_vl.qwen25_vl_bridge
• bridge.models.nemotron_omni
  • bridge.models.nemotron_omni.data
    • bridge.models.nemotron_omni.data.collate_fn
  • bridge.models.nemotron_omni.modeling_nemotron_omni
  • bridge.models.nemotron_omni.inference_inputs
  • bridge.models.nemotron_omni.hf_pretrained
  • bridge.models.nemotron_omni.nemotron_omni_sound
  • bridge.models.nemotron_omni.nemotron_omni_utils
  • bridge.models.nemotron_omni.nemotron_omni_provider
  • bridge.models.nemotron_omni.nemotron_omni_bridge
  • bridge.models.nemotron_omni.modeling_nemotron_omni_llava
  • bridge.models.nemotron_omni.processing_super_vl
• bridge.models.olmoe
  • bridge.models.olmoe.olmoe_provider
  • bridge.models.olmoe.olmoe_bridge
• bridge.models.deepseek
  • bridge.models.deepseek.data
    • bridge.models.deepseek.data.collate_fn
    • bridge.models.deepseek.data.encoding_v4
  • bridge.models.deepseek.common
  • bridge.models.deepseek.deepseek_v4_bridge
  • bridge.models.deepseek.deepseek_v4_step
  • bridge.models.deepseek.deepseek_v3_bridge
  • bridge.models.deepseek.attention
  • bridge.models.deepseek.deepseek_v4_hybrid_provider
• bridge.models.llama
  • bridge.models.llama.llama4_utils
  • bridge.models.llama.llama_bridge
• bridge.models.conversion
  • bridge.models.conversion.quant_mapping
  • bridge.models.conversion.param_mapping
  • bridge.models.conversion.utils
  • bridge.models.conversion.auto_bridge
  • bridge.models.conversion.transformers_compat
  • bridge.models.conversion.peft_bridge
  • bridge.models.conversion.gtp
  • bridge.models.conversion.model_bridge
  • bridge.models.conversion.quantization_utils
  • bridge.models.conversion.quant_bridge
  • bridge.models.conversion.fp8_export
  • bridge.models.conversion.modelopt_utils
  • bridge.models.conversion.mapping_registry
• bridge.models.muse_glimmer
  • bridge.models.muse_glimmer.data
    • bridge.models.muse_glimmer.data.collate_fn
  • bridge.models.muse_glimmer.modeling_muse_glimmer
  • bridge.models.muse_glimmer.muse_glimmer_builder
  • bridge.models.muse_glimmer.muse_glimmer_bridge
  • bridge.models.muse_glimmer.muse_glimmer_config
• bridge.models.minimax_m2
  • bridge.models.minimax_m2.minimax_m2_bridge
  • bridge.models.minimax_m2.minimax_m2_provider
• bridge.models.ministral3
  • bridge.models.ministral3.data
    • bridge.models.ministral3.data.collate_fn
  • bridge.models.ministral3.ministral3_provider
  • bridge.models.ministral3.ministral3_bridge
  • bridge.models.ministral3.modeling_ministral3
• bridge.models.mamba
  • bridge.models.mamba.mamba_builder
  • bridge.models.mamba.mamba_provider
• bridge.models.bailing
  • bridge.models.bailing.modeling_bailing_moe_v2
  • bridge.models.bailing.bailing_moe2_bridge
  • bridge.models.bailing.configuration_bailing_moe_v2

Submodules

• bridge.models.gpt_full_te_layer_autocast_spec
• bridge.models.mla_provider
• bridge.models.gpt_provider
• bridge.models.logit_dtype
• bridge.models.model_provider
• bridge.models.config
• bridge.models.t5_provider
• bridge.models.transformer_config
• bridge.models.distillation_provider

Package Contents

Data

_BAGEL_EXPORTS
__all__

API

bridge.models._BAGEL_EXPORTS

[]

bridge.models.__all__

None