Yoach Lacombe 5da3db3fd5 [Whisper] Fix word-level timestamps with bs>1 or num_beams>1 (#28114) ...

* fix frames

* use smaller chunk length

* correct beam search + tentative stride

* fix whisper word timestamp in batch

* add test batch generation with return token timestamps

* Apply suggestions from code review

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>

* clean a test

* make style + correct typo

* write clearer comments

* explain test in comment

---------

Co-authored-by: sanchit-gandhi <sanchit@huggingface.co>
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>

2023-12-22 12:43:11 +00:00

..

__init__.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_pipelines_audio_classification.py

Allow dict input for audio classification pipeline (#23445)

2023-06-23 13:50:37 +01:00

test_pipelines_automatic_speech_recognition.py

[Whisper] Fix word-level timestamps with bs>1 or num_beams>1 (#28114)

2023-12-22 12:43:11 +00:00

test_pipelines_common.py

Fix RequestCounter to make it more future-proof (#27406)

2023-11-09 18:53:26 +01:00

test_pipelines_conversational.py

[Styling] stylify using ruff (#27144)

2023-11-16 17:43:19 +01:00

test_pipelines_depth_estimation.py

Update fixtures-image-utils (#28080)

2023-12-15 16:58:36 +00:00

test_pipelines_document_question_answering.py

Mark pipeline tests to skip them easily (#21887)

2023-03-02 10:55:36 -05:00

test_pipelines_feature_extraction.py

Mark pipeline tests to skip them easily (#21887)

2023-03-02 10:55:36 -05:00

test_pipelines_fill_mask.py

Fix 2 tests in FillMaskPipelineTests (#27889)

2023-12-08 14:55:29 +01:00

test_pipelines_image_classification.py

Update fixtures-image-utils (#28080)

2023-12-15 16:58:36 +00:00

test_pipelines_image_segmentation.py

Update fixtures-image-utils (#28080)

2023-12-15 16:58:36 +00:00

test_pipelines_image_to_image.py

Add image to image pipeline (#25393)

2023-09-22 19:53:55 +03:00

test_pipelines_image_to_text.py

[Llava] Add Llava to transformers (#27662)

2023-12-07 09:30:47 +01:00

test_pipelines_mask_generation.py

Set usedforsecurity=False in hashlib methods (FIPS compliance) (#27483)

2023-11-16 14:29:53 +00:00

test_pipelines_object_detection.py

Update fixtures-image-utils (#28080)

2023-12-15 16:58:36 +00:00

test_pipelines_question_answering.py

[Styling] stylify using ruff (#27144)

2023-11-16 17:43:19 +01:00

test_pipelines_summarization.py

Proper build() methods for TF (#27794)

2023-12-14 15:17:30 +00:00

test_pipelines_table_question_answering.py

Skip some TQAPipelineTests tests in past CI (#24267)

2023-06-14 14:25:24 +02:00

test_pipelines_text2text_generation.py

Mark pipeline tests to skip them easily (#21887)

2023-03-02 10:55:36 -05:00

test_pipelines_text_classification.py

device agnostic pipelines testing (#27129)

2023-10-31 15:46:31 +01:00

test_pipelines_text_generation.py

Update tiny model summary file (#27388)

2023-11-23 21:00:39 +01:00

test_pipelines_text_to_audio.py

Enrich TTS pipeline parameters naming (#26473)

2023-11-02 17:06:56 +00:00

test_pipelines_token_classification.py

[Styling] stylify using ruff (#27144)

2023-11-16 17:43:19 +01:00

test_pipelines_translation.py

Mark pipeline tests to skip them easily (#21887)

2023-03-02 10:55:36 -05:00

test_pipelines_video_classification.py

Mark pipeline tests to skip them easily (#21887)

2023-03-02 10:55:36 -05:00

test_pipelines_visual_question_answering.py

device agnostic pipelines testing (#27129)

2023-10-31 15:46:31 +01:00

test_pipelines_zero_shot_audio_classification.py

Enable ZeroShotAudioClassificationPipelineTests::test_small_model_pt (#24882)

2023-07-18 15:08:53 +02:00

test_pipelines_zero_shot_image_classification.py

🚨🚨🚨 Fix rescale ViVit Efficientnet (#25174)

2023-07-28 19:52:51 +01:00

test_pipelines_zero_shot_object_detection.py

Mark pipeline tests to skip them easily (#21887)

2023-03-02 10:55:36 -05:00

test_pipelines_zero_shot.py

Update tiny models and a few fixes (#22928)

2023-04-24 14:45:22 +02:00