Adds OWLViT to models exportable with ONNX (#18588)

* onnx conversion for owlvit * .T to .t() * dynamic shapes for pixel values
2022-08-30 18:00:59 +05:30
parent b83796ded7
commit 46d0e26a27
6 changed files with 62 additions and 3 deletions
--- a/tests/onnx/test_onnx_v2.py
+++ b/tests/onnx/test_onnx_v2.py
@@ -205,6 +205,7 @@ PYTORCH_EXPORT_MODELS = {
    ("layoutlm", "microsoft/layoutlm-base-uncased"),
    ("layoutlmv3", "microsoft/layoutlmv3-base"),
    ("levit", "facebook/levit-128S"),
+    ("owlvit", "google/owlvit-base-patch32"),
    ("vit", "google/vit-base-patch16-224"),
    ("deit", "facebook/deit-small-patch16-224"),
    ("beit", "microsoft/beit-base-patch16-224"),