chunk_content:       input_features=batch["input_features"].to("cuda"),
                      decoder_input_ids=batch["labels"][:, :4].to("cuda"),
                      max_new_tokens=255,
                  )
                  .cpu()
                  .numpy()
              )
              labels = batch["labels"].cpu().numpy()
              labels = np.where(labels != -100, labels, tokenizer.pad_token_id)
              decoded_preds = tokenizer.batch_decode(generated_tokens, 