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(
    self,
    tokenizer,
    processor,
    return_tensors='pt',
    chat_template=None,
    packing=False,
    max_images=-1,
    padding=True,
    pad_to_multiple_of=None,
)

Collator for multi-modal chat messages