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	Add WhisperModel to transformers (#19166)	2022-10-05 22:28:31 +02:00
test_feature_extraction_whisper.py	[Whisper] Use torch for stft if available (#26119)	2023-12-21 11:04:05 +00:00
test_modeling_flax_whisper.py	[Styling] stylify using ruff (#27144)	2023-11-16 17:43:19 +01:00
test_modeling_tf_whisper.py	More TF fixes (#28081)	2023-12-18 15:26:03 +00:00
test_modeling_whisper.py	[Whisper] Fix word-level timestamps with bs>1 or num_beams>1 (#28114)	2023-12-22 12:43:11 +00:00
test_processor_whisper.py	feat: Whisper prompting (#22496)	2023-05-19 09:33:11 +01:00
test_tokenization_whisper.py	[Styling] stylify using ruff (#27144)	2023-11-16 17:43:19 +01:00