Younes Belkada 914289ac4b [pipeline] Fix str device issue (#24396) ...

* fix str device issue

* fixup

* adapt from suggestions

* forward contrib credits from suggestions

* better fix

* added backward compatibility for older PT versions

* final fixes

* oops

* Attempting something with less branching.

---------

Co-authored-by: amyeroberts <amyeroberts@users.noreply.github.com>
Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com>

2023-06-26 13:58:36 +02: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 word-level timestamps to Whisper (#23205)

2023-06-21 17:48:21 +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 Pipeline CI OOM issue (#24124)

2023-06-09 16:49:02 +02: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

Fix Pipeline CI OOM issue (#24124)

2023-06-09 16:49:02 +02: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

Really fix quality due to ruff release

2023-03-22 20:56:22 -04: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_text2text_generation.py

Mark pipeline tests to skip them easily (#21887)

2023-03-02 10:55:36 -05: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

Fix (skip) a pipeline test for RwkvModel (#23444)

2023-05-18 14:54:23 +02: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

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

🔥py38 + torch 2 🔥🔥🔥🚀 (#22204)

2023-03-16 22:59:23 +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