Adds GroupViT to models exportable with ONNX (#18628)

* groupvit to onnx

* dynamic shape for pixel values dim
This commit is contained in:
Dhruv Karan
2022-08-30 18:01:35 +05:30
committed by GitHub
parent 46d0e26a27
commit 220da3b8a1
6 changed files with 58 additions and 2 deletions

View File

@@ -204,6 +204,7 @@ PYTORCH_EXPORT_MODELS = {
("xlm-roberta", "xlm-roberta-base"),
("layoutlm", "microsoft/layoutlm-base-uncased"),
("layoutlmv3", "microsoft/layoutlmv3-base"),
("groupvit", "nvidia/groupvit-gcc-yfcc"),
("levit", "facebook/levit-128S"),
("owlvit", "google/owlvit-base-patch32"),
("vit", "google/vit-base-patch16-224"),