transformers/tests/pipelines
Sylvain Gugger 42ad693b7b
Regression pipeline device (#22190)
* Fix regression in pipeline when device=-1 is passed

* Add regression test
2023-03-15 14:13:38 -04: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 Update AudioClassificationPipelineTests::test_small_model_pt for PT 2.0.0 (#22023) 2023-03-08 13:56:47 +01:00
test_pipelines_automatic_speech_recognition.py Refactor whisper asr pipeline to include language too. (#21427) 2023-03-02 18:12:19 +01:00
test_pipelines_common.py Regression pipeline device (#22190) 2023-03-15 14:13:38 -04:00
test_pipelines_conversational.py Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05: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 Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05:00
test_pipelines_image_classification.py Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05:00
test_pipelines_image_segmentation.py Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05:00
test_pipelines_image_to_text.py Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05: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 Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05: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 Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05:00
test_pipelines_text_classification.py Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05:00
test_pipelines_text_generation.py Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05: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 Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05: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 Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05:00
test_pipelines_zero_shot_audio_classification.py Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05:00
test_pipelines_zero_shot_image_classification.py Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05: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 Mark pipeline tests to skip them easily (#21887) 2023-03-02 10:55:36 -05:00