[Whisper + beam search] fix usage of beam_indices (#38259)

* tmp * fix test_tiny_token_timestamp_batch_generation * better comments * test * comments * Apply suggestions from code review Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com> --------- Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com>
2025-05-23 11:05:44 +01:00
parent 3e960e032d
commit a6b51e7341
2 changed files with 27 additions and 18 deletions
--- a/tests/models/whisper/test_modeling_whisper.py
+++ b/tests/models/whisper/test_modeling_whisper.py
@@ -2155,7 +2155,6 @@ class WhisperModelIntegrationTests(unittest.TestCase):

        # task id and lang id prompts should not have timestamp tokens
        self.assertEqual(generate_outputs["sequences"].shape[-1] - 2, generate_outputs["token_timestamps"].shape[-1])
-
        self.assertEqual(len(generate_outputs["sequences"]), num_return_sequences * num_samples)

    @slow