chunk_content: ut_ids"].masked_fill(labels_batch.attention_mask.ne(1), -100)
  
          if (labels[:, 0] == self.processor.tokenizer.bos_token_id).all().cpu().item():
              labels = labels[:, 1:]
  
          batch["labels"] = labels
  
          return batch
  
  
  data_collator = DataCollatorSpeechSeq2SeqWithPadding(processor=processor)
  
  Train
  	
  Now that the dataset is ready, you can turn your attention to the model. Start by loading the pretrained openai/whisper-la