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/docs/source/en/serialization.mdx
+++ b/docs/source/en/serialization.mdx
@@ -70,6 +70,7 @@ Ready-made configurations include the following architectures:
 - OpenAI GPT-2
 - PLBart
 - RoBERTa
+- RoFormer
 - T5
 - TAPEX
 - ViT