Add onnx config for RoFormer (#16861)

* add roformer onnx config
2022-04-26 10:51:15 -04:00
parent 8afaaa26f5
commit aaee4038c3
5 changed files with 37 additions and 2 deletions
--- a/tests/onnx/test_onnx_v2.py
+++ b/tests/onnx/test_onnx_v2.py
@@ -179,6 +179,7 @@ PYTORCH_EXPORT_MODELS = {
    ("distilbert", "distilbert-base-cased"),
    ("electra", "google/electra-base-generator"),
    ("roberta", "roberta-base"),
+    ("roformer", "junnyu/roformer_chinese_base"),
    ("xlm-roberta", "xlm-roberta-base"),
    ("layoutlm", "microsoft/layoutlm-base-uncased"),
    ("vit", "google/vit-base-patch16-224"),