`bridge.models.muse_glimmer.data.collate_fn`#

Muse Glimmer multimodal SFT collator.

Module Contents#

Functions#

muse_glimmer_collate_fn

Collate Muse Glimmer conversations and their THW-grid media tensors.

Data#

`MUSE_GLIMMER_ASSISTANT_START`
`MUSE_GLIMMER_TURN_END`
`__all__`

API#

bridge.models.muse_glimmer.data.collate_fn.MUSE_GLIMMER_ASSISTANT_START#: ‘<|start|>assistant to=user<|message|>’

bridge.models.muse_glimmer.data.collate_fn.MUSE_GLIMMER_TURN_END#: ‘<|eot|>’

bridge.models.muse_glimmer.data.collate_fn.muse_glimmer_collate_fn( examples: list[dict[str, Any]], processor: Any, *, 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, ) → dict[str, Any]#: Collate Muse Glimmer conversations and their THW-grid media tensors.

bridge.models.muse_glimmer.data.collate_fn.__all__#: [‘muse_glimmer_collate_fn’]