`bridge.models.kimi.kimi_bridge`#

Module Contents#

Classes#

Megatron Bridge for Kimi K2.

API#

class bridge.models.kimi.kimi_bridge.KimiK2Bridge#

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

Megatron Bridge for Kimi K2.

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

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