riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 10:41:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity
ec6cd7633f
transformers/tests/pipelines
History
Sylvain Gugger dc9147ff36
Custom pipeline (#18079) 
* Initial work

* More work

* Add tests for custom pipelines on the Hub

* Protect import

* Make the test work for TF as well

* Last PyTorch specific bit

* Add documentation

* Style

* Title in toc

* Bad names!

* Update docs/source/en/add_new_pipeline.mdx

Co-authored-by: Lysandre Debut <lysandre.debut@reseau.eseo.fr>

* Auto stash before merge of "custom_pipeline" and "origin/custom_pipeline"

* Address review comments

* Address more review comments

* Update src/transformers/pipelines/__init__.py

Co-authored-by: Lysandre Debut <lysandre.debut@reseau.eseo.fr>

Co-authored-by: Lysandre Debut <lysandre.debut@reseau.eseo.fr>
2022-07-19 12:02:35 +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 [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_pipelines_automatic_speech_recognition.py Black preview (#17217) 2022-05-12 16:25:55 -04:00
test_pipelines_common.py Custom pipeline (#18079) 2022-07-19 12:02:35 +02:00
test_pipelines_conversational.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_pipelines_feature_extraction.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_pipelines_fill_mask.py Running a pipeline of float16. (#17637) 2022-06-09 19:04:42 +02:00
test_pipelines_image_classification.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_pipelines_image_segmentation.py Fix image segmentation and object detection pipeline tests (#18100) 2022-07-11 12:41:56 -04:00
test_pipelines_object_detection.py Fix image segmentation and object detection pipeline tests (#18100) 2022-07-11 12:41:56 -04:00
test_pipelines_question_answering.py Adding batch_size test to QA pipeline. (#17330) 2022-05-19 14:28:12 -04:00
test_pipelines_summarization.py Add LongT5 model (#16792) 2022-06-13 22:36:58 +02:00
test_pipelines_table_question_answering.py Black preview (#17217) 2022-05-12 16:25:55 -04:00
test_pipelines_text_classification.py Fixing a regression with return_all_scores introduced in #17606 (#17906) 2022-06-28 17:24:45 -04:00
test_pipelines_text_generation.py Adding support for device_map directly in pipeline(..) function. (#17902) 2022-07-15 15:54:26 +02:00
test_pipelines_text2text_generation.py Fixing return type tensor with num_return_sequences>1. (#16828) 2022-04-20 16:11:51 +02:00
test_pipelines_token_classification.py Fix some typos. (#17560) 2022-07-11 05:00:13 -04:00
test_pipelines_translation.py Black preview (#17217) 2022-05-12 16:25:55 -04:00
test_pipelines_visual_question_answering.py Add Visual Question Answering (VQA) pipeline (#17286) 2022-06-13 07:49:44 -04:00
test_pipelines_zero_shot_image_classification.py The tests were not updated after the addition of torch.diag (#15890) 2022-03-03 15:33:49 +01:00
test_pipelines_zero_shot.py Black preview (#17217) 2022-05-12 16:25:55 -04:00