transformers/tests/pipelines
NielsRogge f278ef20ed
[Nougat] Fix pipeline (#28242)
* Fix pipeline

* Remove print statements

* Address comments

* Address issue

* Remove unused imports
2024-02-12 10:21:15 +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 Correct wav2vec2-bert inputs_to_logits_ratio (#28821) 2024-02-05 13:14:47 +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 Enable multi-label image classification in pipeline (#28433) 2024-01-11 10:29:38 +00:00
test_pipelines_image_feature_extraction.py Image Feature Extraction pipeline (#28216) 2024-02-05 14:50:07 +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 [Nougat] Fix pipeline (#28242) 2024-02-12 10:21:15 +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_text_classification.py device agnostic pipelines testing (#27129) 2023-10-31 15:46:31 +01:00
test_pipelines_text_generation.py ⚠️ Raise Exception when trying to generate 0 tokens ⚠️ (#28621) 2024-02-07 13:42:01 +01:00
test_pipelines_text_to_audio.py Enrich TTS pipeline parameters naming (#26473) 2023-11-02 17:06:56 +00:00
test_pipelines_text2text_generation.py Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05: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 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
test_pipelines_zero_shot.py Update tiny models and a few fixes (#22928) 2023-04-24 14:45:22 +02:00