MyCollator() Only Processes batch[0]

[zlin0]

Hi @shangeth, thanks for opensource your code! May I inquire about another question? In the MyCollator() function, you only handle batch[0]. Is there a specific reason for that?
https://github.com/skit-ai/SpeechLLM/blob/main/dataset.py#L18
waveform, pre_speech_prompt, post_speech_prompt, output_prompt, complete_prompt = batch[0]

Since this collator function is used by both training and validation, this row means that only the first sample in each mini-batch will be utilized. Or did I miss or misunderstand anything? Thanks!