Rename second input dimension from "sequence" to "num_channels" for CV models (#17976)

2022-08-18 15:13:54 +02:00
parent 780253ce3d
commit 76454b08c8
10 changed files with 11 additions and 10 deletions
--- a/tests/onnx/test_onnx_v2.py
+++ b/tests/onnx/test_onnx_v2.py
@@ -199,6 +199,7 @@ PYTORCH_EXPORT_MODELS = {
    ("roformer", "junnyu/roformer_chinese_base"),
    ("squeezebert", "squeezebert/squeezebert-uncased"),
    ("mobilebert", "google/mobilebert-uncased"),
+    ("mobilevit", "apple/mobilevit-small"),
    ("xlm", "xlm-clm-ende-1024"),
    ("xlm-roberta", "xlm-roberta-base"),
    ("layoutlm", "microsoft/layoutlm-base-uncased"),