Add RemBERT ONNX config (#20520)

* rembert onnx config

* formatting

Co-authored-by: Ho <erincho@bcd0745f972b.ant.amazon.com>
This commit is contained in:
Erin
2022-12-05 08:39:09 -08:00
committed by GitHub
parent afe2a466bb
commit 87282cb73c
5 changed files with 39 additions and 2 deletions

View File

@@ -210,6 +210,7 @@ PYTORCH_EXPORT_MODELS = {
("owlvit", "google/owlvit-base-patch32"),
("perceiver", "hf-internal-testing/tiny-random-PerceiverModel", ("masked-lm", "sequence-classification")),
("perceiver", "hf-internal-testing/tiny-random-PerceiverModel", ("image-classification",)),
("rembert", "google/rembert"),
("resnet", "microsoft/resnet-50"),
("roberta", "hf-internal-testing/tiny-random-RobertaModel"),
("roformer", "hf-internal-testing/tiny-random-RoFormerModel"),