def METHOD_NAME(self, text_column_names: Optional[List] = None) -> Dict:
    """
    Collate multimodal features into a batch.
    This function will be used when creating Pytorch DataLoader.
    Returns
    -------
    A dictionary containing one model's collator function for multimodal data.
    """
    fn = {}
    if self.prefix == NER_TEXT:
        fn.update(
            {
                self.text_token_ids_key: PadCollator(pad_val=self.tokenizer.pad_token_id),
                self.text_valid_length_key: StackCollator(),
                self.text_segment_ids_key: PadCollator(pad_val=0),
                self.text_token_word_mapping_key: PadCollator(pad_val=0),
                self.text_word_offsets_key: PadCollator(pad_val=0),
                self.label_key: StackCollator(),
            }
        )
    return fn