Added XLM onnx config (#17030)

* Add onnx configuration for xlm

* Add supported features for xlm

* Add xlm to models exportable with onnx

* Add xlm architecture to test file

* Modify docs

* Make code quality fixes
This commit is contained in:
Ritik Nandwal
2022-05-31 18:56:06 +05:30
committed by GitHub
parent 567d9c061d
commit 5af38953bb
5 changed files with 35 additions and 2 deletions

View File

@@ -181,6 +181,7 @@ PYTORCH_EXPORT_MODELS = {
("roberta", "roberta-base"),
("roformer", "junnyu/roformer_chinese_base"),
("mobilebert", "google/mobilebert-uncased"),
("xlm", "xlm-clm-ende-1024"),
("xlm-roberta", "xlm-roberta-base"),
("layoutlm", "microsoft/layoutlm-base-uncased"),
("vit", "google/vit-base-patch16-224"),