Adds IBERT to models exportable with ONNX (#14868)

* Add IBertOnnxConfig and tests

* add all the supported features for IBERT and remove outputs in IbertOnnxConfig

* use OnnxConfig

* fix codestyle

* remove serialization.rst

* codestyle
This commit is contained in:
Virus
2022-01-11 14:17:08 +03:00
committed by GitHub
parent efb35a4107
commit c4fa908fa9
5 changed files with 29 additions and 2 deletions

View File

@@ -171,6 +171,7 @@ class OnnxConfigWithPastTestCaseV2(TestCase):
PYTORCH_EXPORT_MODELS = {
("albert", "hf-internal-testing/tiny-albert"),
("bert", "bert-base-cased"),
("ibert", "kssteven/ibert-roberta-base"),
("camembert", "camembert-base"),
("distilbert", "distilbert-base-cased"),
# ("longFormer", "longformer-base-4096"),