transformers/tests/pipelines
Matt 3b44d2f042
Image pipelines spec compliance (#33899)
* Update many similar visual pipelines

* Add input tests

* Add ImageToText as well

* Add output tests

* Add output tests

* Add output tests

* OutputElement -> Output

* Correctly test elements

* make fixup

* fix typo in the task list

* Fix VQA testing

* Add copyright to image_classification.py

* Revert changes to VQA pipeline because outputs have differences - will move to another PR

* make fixup

* Remove deprecation warnings
2024-10-08 13:34:28 +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 Make ASR pipeline compliant with Hub spec + add tests (#33769) 2024-10-01 18:15:04 +01:00
test_pipelines_automatic_speech_recognition.py Make ASR pipeline compliant with Hub spec + add tests (#33769) 2024-10-01 18:15:04 +01:00
test_pipelines_common.py Pipeline: no side-effects on model.config and model.generation_config 🔫 (#33480) 2024-09-18 15:43:06 +01:00
test_pipelines_depth_estimation.py Image pipelines spec compliance (#33899) 2024-10-08 13:34:28 +01:00
test_pipelines_document_question_answering.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_feature_extraction.py fix: Fixed raising TypeError instead of ValueError for invalid type (#32111) 2024-07-22 17:46:17 +01:00
test_pipelines_fill_mask.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_image_classification.py Image pipelines spec compliance (#33899) 2024-10-08 13:34:28 +01:00
test_pipelines_image_feature_extraction.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_image_segmentation.py Image pipelines spec compliance (#33899) 2024-10-08 13:34:28 +01:00
test_pipelines_image_to_image.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_image_to_text.py Image pipelines spec compliance (#33899) 2024-10-08 13:34:28 +01:00
test_pipelines_mask_generation.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_object_detection.py Image pipelines spec compliance (#33899) 2024-10-08 13:34:28 +01:00
test_pipelines_question_answering.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_summarization.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_table_question_answering.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_text_classification.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_text_generation.py Compile compatibilty for decoder-only models (#32617) 2024-09-09 10:59:04 +02:00
test_pipelines_text_to_audio.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_text2text_generation.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_token_classification.py enable low-precision pipeline (#31625) 2024-09-20 16:43:30 -07:00
test_pipelines_translation.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_video_classification.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_visual_question_answering.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_zero_shot_audio_classification.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_zero_shot_image_classification.py Image pipelines spec compliance (#33899) 2024-10-08 13:34:28 +01:00
test_pipelines_zero_shot_object_detection.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_zero_shot.py enable low-precision pipeline (#31625) 2024-09-20 16:43:30 -07:00