`bridge.models.ministral3.data.collate_fn`#

Ministral 3 VL collator implementations.

Module Contents#

`_has_generation_chat_template`
`_default_ministral3_assistant_mask_boundary_config`
`ministral3_collate_fn`	Collate function for Ministral 3 VL model.

`MISTRAL3_ASSISTANT_START`
`MISTRAL3_ASSISTANT_END`

bridge.models.ministral3.data.collate_fn.MISTRAL3_ASSISTANT_START#: ‘[/INST]’

bridge.models.ministral3.data.collate_fn._has_generation_chat_template(processor) → bool#

bridge.models.ministral3.data.collate_fn._default_ministral3_assistant_mask_boundary_config( processor, ) → megatron.bridge.data.conversation_processing.AssistantMaskBoundaryConfig#

bridge.models.ministral3.data.collate_fn.ministral3_collate_fn( examples: list, processor, *, visual_keys: object = None, min_pixels: int | None = None, max_pixels: int | None = None, sequence_length: int | None = None, pad_to_max_length: bool = False, pad_to_multiple_of: int = 128, enable_in_batch_packing: bool = False, in_batch_packing_pad_to_multiple_of: int = 1, assistant_mask_boundary_config: megatron.bridge.data.conversation_processing.AssistantMaskBoundaryConfig | None = None, ) → dict[str, torch.Tensor]#: Collate function for Ministral 3 VL model.