transformers/tests/pipelines
Matt 3b3024da70
TF port of ESM (#19587)
* Partial TF port for ESM model

* Add ESM-TF tests

* Add the various imports for TF-ESM

* TF weight conversion almost ready

* Stop ignoring the decoder weights in PT

* Add tests and lots of fixes

* fix-copies

* Fix imports, add model docs

* Add get_vocab() to tokenizer

* Fix vocab links for pretrained files

* Allow multiple inputs with a sep

* Use EOS as SEP token because ESM vocab lacks SEP

* Correctly return special tokens mask from ESM tokenizer

* make fixup

* Stop testing unsupported embedding resizing

* Handle TF bias correctly

* Skip all models with slow tokenizers in the token classification test

* Fixing the batch/unbatcher of pipelines to accomodate the `None` being

passed around.

* Fixing pipeline bug caused by slow tokenizer  being different.

* Update src/transformers/models/esm/modeling_tf_esm.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Update src/transformers/models/esm/modeling_tf_esm.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Update src/transformers/models/esm/modeling_tf_esm.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Update set_input_embeddings and the copyright notices

Co-authored-by: Your Name <you@example.com>
Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com>
Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>
2022-10-17 14:16:16 +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 Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_automatic_speech_recognition.py Improve error messaging for ASR pipeline. (#19570) 2022-10-14 17:12:21 +02:00
test_pipelines_common.py Fix pipeline predict transform methods (#19657) 2022-10-17 09:06:20 -04:00
test_pipelines_conversational.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_depth_estimation.py Add depth estimation pipeline (#18618) 2022-10-12 08:54:20 -04:00
test_pipelines_document_question_answering.py Implement multiple span support for DocumentQuestionAnswering (#19204) 2022-10-11 10:47:55 -04:00
test_pipelines_feature_extraction.py Added tokenize keyword arguments to feature extraction pipeline (#19382) 2022-10-11 12:54:41 -04:00
test_pipelines_fill_mask.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_image_classification.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_image_segmentation.py Cast masks to np.unit8 before converting to PIL.Image.Image (#19616) 2022-10-14 09:30:45 -04:00
test_pipelines_image_to_text.py Fix ImageToTextPipelineTests.test_small_model_tf (#19565) 2022-10-14 16:29:54 +02:00
test_pipelines_object_detection.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_question_answering.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_summarization.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_table_question_answering.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_text_classification.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_text_generation.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_text2text_generation.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_token_classification.py TF port of ESM (#19587) 2022-10-17 14:16:16 +01:00
test_pipelines_translation.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_visual_question_answering.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_zero_shot_image_classification.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_zero_shot_object_detection.py Remove ref to is_pipeline_test 2022-10-07 21:38:07 -04:00
test_pipelines_zero_shot.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00