transformers/tests/pipelines
Pavel Iakubovskii 48461c0fe2
Make pipeline able to load processor (#32514)
* Refactor get_test_pipeline

* Fixup

* Fixing tests

* Add processor loading in tests

* Restructure processors loading

* Add processor to the pipeline

* Move model loading on tom of the test

* Update `get_test_pipeline`

* Fixup

* Add class-based flags for loading processors

* Change `is_pipeline_test_to_skip` signature

* Skip t5 failing test for slow tokenizer

* Fixup

* Fix copies for T5

* Fix typo

* Add try/except for tokenizer loading (kosmos-2 case)

* Fixup

* Llama not fails for long generation

* Revert processor pass in text-generation test

* Fix docs

* Switch back to json file for image processors and feature extractors

* Add processor type check

* Remove except for tokenizers

* Fix docstring

* Fix empty lists for tests

* Fixup

* Fix load check

* Ensure we have non-empty test cases

* Update src/transformers/pipelines/__init__.py

Co-authored-by: Lysandre Debut <hi@lysand.re>

* Update src/transformers/pipelines/base.py

Co-authored-by: Lysandre Debut <hi@lysand.re>

* Rework comment

* Better docs, add note about pipeline components

* Change warning to error raise

* Fixup

* Refine pipeline docs

---------

Co-authored-by: Lysandre Debut <hi@lysand.re>
2024-10-09 16:46:11 +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 pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_automatic_speech_recognition.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +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 Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_document_question_answering.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_feature_extraction.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_fill_mask.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_image_classification.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_image_feature_extraction.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_image_segmentation.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +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 Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_mask_generation.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_object_detection.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_question_answering.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_summarization.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +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 Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_text_generation.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_text_to_audio.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_text2text_generation.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_token_classification.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_translation.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_video_classification.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_visual_question_answering.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +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 Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_zero_shot.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00