transformers/tests/pipelines
Andreas Madsen b4b613b102
Implement Roberta PreLayerNorm (#20305)
* Copy RoBERTa

* formatting

* implement RoBERTa with prelayer normalization

* update test expectations

* add documentation

* add convertion script for DinkyTrain weights

* update checkpoint repo

Unfortunately the original checkpoints assumes a hacked roberta model

* add to RoBERTa-PreLayerNorm docs to toc

* run utils/check_copies.py

* lint files

* remove unused import

* fix check_repo reporting wrongly a test is missing

* fix import error, caused by rebase

* run make fix-copies

* add RobertaPreLayerNormConfig to ROBERTA_EMBEDDING_ADJUSMENT_CONFIGS

* Fix documentation <Facebook> -> Facebook

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

* fixup: Fix documentation <Facebook> -> Facebook

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

* Add missing Flax header

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

* expected_slice -> EXPECTED_SLICE

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

* update copies after rebase

* add missing copied from statements

* make fix-copies

* make prelayernorm explicit in code

* fix checkpoint path for the original implementation

* add flax integration tests

* improve docs

* update utils/documentation_tests.txt

* lint files

* Remove Copyright notice

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* make fix-copies

* Remove EXPECTED_SLICE calculation comments

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2022-12-19 09:30:17 +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 Fix AutomaticSpeechRecognitionPipelineTests.run_pipeline_test (#20597) 2022-12-06 15:48:49 +01:00
test_pipelines_common.py Implement Roberta PreLayerNorm (#20305) 2022-12-19 09:30:17 +01: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 A few CI fixes for DocumentQuestionAnsweringPipeline (#19584) 2022-10-17 15:35:27 +02:00
test_pipelines_feature_extraction.py [Pipeline] skip feature extraction test if in IMAGE_PROCESSOR_MAPPING (#20790) 2022-12-16 12:46:58 +01:00
test_pipelines_fill_mask.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_image_classification.py Specify TF framework explicitly in more pipeline tests (#19748) 2022-10-19 16:24:03 +02:00
test_pipelines_image_segmentation.py Make ImageSegmentationPipelineTests less flaky (#20147) 2022-11-15 09:14:55 +01:00
test_pipelines_image_to_text.py fix image2test args forwarding (#19648) 2022-10-24 09:49:24 -04:00
test_pipelines_object_detection.py Fix object detection2 (#20798) 2022-12-16 13:25:36 +01:00
test_pipelines_question_answering.py Clean up deprecation warnings (#19654) 2022-10-18 13:34:47 -04:00
test_pipelines_summarization.py Change a logic in pipeline test regarding TF (#20710) 2022-12-13 13:42:36 +01:00
test_pipelines_table_question_answering.py Add require_torch to 2 pipeline tests (#20585) 2022-12-05 16:06:39 +01:00
test_pipelines_text_classification.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00
test_pipelines_text_generation.py [Pipeline] fix failing bloom pipeline test (#20778) 2022-12-15 18:46:00 +01: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_video_classification.py Add video classification pipeline (#20151) 2022-12-08 16:22:43 -05: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 [Proposal] Breaking change zero-shot-object-detection for improved consistency. (#20280) 2022-11-18 15:57:28 +01:00
test_pipelines_zero_shot.py Rework pipeline tests (#19366) 2022-10-07 18:01:58 -04:00