Adds OWLViT to models exportable with ONNX (#18588)

* onnx conversion for owlvit

* .T to .t()

* dynamic shapes for pixel values
This commit is contained in:
Dhruv Karan
2022-08-30 18:00:59 +05:30
committed by GitHub
parent b83796ded7
commit 46d0e26a27
6 changed files with 62 additions and 3 deletions

View File

@@ -83,6 +83,7 @@ Ready-made configurations include the following architectures:
- MobileViT
- MT5
- OpenAI GPT-2
- OWL-ViT
- Perceiver
- PLBart
- ResNet