Squash commits (#17981)

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
2022-07-06 14:11:48 +02:00
parent f681437203
commit 22edb68d49
5 changed files with 36 additions and 2 deletions
--- a/tests/onnx/test_onnx_v2.py
+++ b/tests/onnx/test_onnx_v2.py
@@ -202,6 +202,7 @@ PYTORCH_EXPORT_MODELS = {
    ("data2vec-text", "facebook/data2vec-text-base"),
    ("perceiver", "deepmind/language-perceiver", ("masked-lm", "sequence-classification")),
    ("perceiver", "deepmind/vision-perceiver-conv", ("image-classification",)),
+    ("yolos", "hustvl/yolos-tiny"),
 }

 PYTORCH_EXPORT_WITH_PAST_MODELS = {