utils.collators.mm_chat

utils.collators.mm_chat

Collators for multi-modal chat messages and packing

Classes

Name	Description
MultiModalChatDataCollator	Collator for multi-modal chat messages

MultiModalChatDataCollator

utils.collators.mm_chat.MultiModalChatDataCollator(
    tokenizer,
    processing_strategy,
    packing=False,
    return_tensors='pt',
    padding=True,
    pad_to_multiple_of=None,
)

Collator for multi-modal chat messages