model/data_collator.py

from transformers import DataCollatorSpeechSeq2SeqWithPadding

def get_data_collator(processor):
    return DataCollatorSpeechSeq2SeqWithPadding(
        processor=processor,
        padding=True
    )