Add RemBERT ONNX config (#20520)
* rembert onnx config * formatting Co-authored-by: Ho <erincho@bcd0745f972b.ant.amazon.com>
This commit is contained in:
@@ -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"),
|
||||
|
||||
Reference in New Issue
Block a user