transformers/tests/pipelines
Matt bdbcd5d482
Fix and re-enable ConversationalPipeline tests (#26907)
* Fix and re-enable conversationalpipeline tests

* Fix the batch test so the change only applies to conversational pipeline
2023-10-19 12:04:25 +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 Add many missing spaces in adjacent strings (#26751) 2023-10-12 10:28:40 +02:00
test_pipelines_common.py [pipeline] Fix str device issue (#24396) 2023-06-26 13:58:36 +02:00
test_pipelines_conversational.py Fix and re-enable ConversationalPipeline tests (#26907) 2023-10-19 12:04:25 +01:00
test_pipelines_depth_estimation.py Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05: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 Add tokenizer kwargs to fill mask pipeline. (#26234) 2023-10-03 10:25:10 +02:00
test_pipelines_image_classification.py 🚨🚨 🚨🚨 [Tokenizer] attemp to fix add_token issues🚨🚨 🚨🚨 (#23909) 2023-09-18 20:28:36 +02:00
test_pipelines_image_segmentation.py Save image_processor while saving pipeline (ImageSegmentationPipeline) (#25884) 2023-08-31 16:08:20 +02: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 Skip test_conditional_generation_pt_pix2struct in Past CI (torch < 1.11) (#24417) 2023-06-22 15:34:13 +02:00
test_pipelines_mask_generation.py Update tiny models for pipeline testing. (#24364) 2023-06-20 14:43:10 +02:00
test_pipelines_object_detection.py Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05:00
test_pipelines_question_answering.py Update tiny models and a few fixes (#22928) 2023-04-24 14:45:22 +02:00
test_pipelines_summarization.py Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05: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 Update tiny models and a few fixes (#22928) 2023-04-24 14:45:22 +02:00
test_pipelines_text_generation.py Llama/GPTNeoX: add RoPE scaling (#24653) 2023-07-13 16:47:30 +01:00
test_pipelines_text_to_audio.py Update tiny model information and pipeline tests (#26285) 2023-09-25 18:08:12 +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_token_classification.py Update tiny models and a few fixes (#22928) 2023-04-24 14:45:22 +02: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 Add Blip2 model in VQA pipeline (#25532) 2023-08-30 14:16:16 +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 🚨🚨🚨 Fix rescale ViVit Efficientnet (#25174) 2023-07-28 19:52:51 +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