transformers/tests/pipelines
Kamil Akesbi eb1a77bbb0
Using assistant in AutomaticSpeechRecognitionPipeline with different encoder size (#30637)
* fiw input to generate in pipeline

* fixup

* pass input_features to generate with assistant

* error if model and assistant with different enc size

* fix

* apply review suggestions

* use self.config.is_encoder_decoder

* pass inputs to generate directly

* add slow tests

* Update src/transformers/generation/utils.py

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

* Update tests/pipelines/test_pipelines_automatic_speech_recognition.py

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

* Update tests/pipelines/test_pipelines_automatic_speech_recognition.py

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

* Update tests/pipelines/test_pipelines_automatic_speech_recognition.py

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

* Update tests/pipelines/test_pipelines_automatic_speech_recognition.py

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

* Update tests/pipelines/test_pipelines_automatic_speech_recognition.py

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

* apply review

* Update src/transformers/generation/utils.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Update tests/pipelines/test_pipelines_automatic_speech_recognition.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* apply code review

* update attributes encoder_xyz to check

* Update src/transformers/generation/utils.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Update src/transformers/generation/utils.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Update src/transformers/generation/utils.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* add slow test

* solve conflicts

---------

Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>
2024-05-23 09:59:38 +01: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 Using assistant in AutomaticSpeechRecognitionPipeline with different encoder size (#30637) 2024-05-23 09:59:38 +01:00
test_pipelines_common.py fix for custom pipeline configuration (#29004) 2024-05-20 11:38:32 +02: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 Fix all torch pipeline failures except one (#30290) 2024-04-18 10:35:43 +02: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 Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
test_pipelines_image_classification.py Enable multi-label image classification in pipeline (#28433) 2024-01-11 10:29:38 +00:00
test_pipelines_image_feature_extraction.py [`pipeline`] Add pool option to image feature extraction pipeline (#28985) 2024-02-20 20:22:08 +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 [tests] fix the wrong output in `ImageToTextPipelineTests.test_conditional_generation_llava` (#29975) 2024-04-01 13:08:39 +02: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 Add chat templating support for KeyDataset in text-generation pipeline (#30558) 2024-04-30 19:51:41 +01:00
test_pipelines_text_to_audio.py if output is tuple like facebook/hf-seamless-m4t-medium, waveform is … (#29722) 2024-04-05 09:26:44 +02:00
test_pipelines_token_classification.py Update all references to canonical models (#29001) 2024-02-16 08:16:58 +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 Replace 'decord' with 'av' in VideoClassificationPipeline (#29747) 2024-03-26 10:12:24 +00:00
test_pipelines_visual_question_answering.py device agnostic pipelines testing (#27129) 2023-10-31 15:46:31 +01:00
test_pipelines_zero_shot.py Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
test_pipelines_zero_shot_audio_classification.py Fix all torch pipeline failures except one (#30290) 2024-04-18 10:35:43 +02:00
test_pipelines_zero_shot_image_classification.py Add SigLIP (#26522) 2024-01-08 18:17:16 +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