Remove a failing ONNX test (huggingface#23011)

fix Co-authored-by: ydshieh <[email protected]>
gojiteji · Jun 5, 2023 · 991ad06 · 991ad06
1 parent fb253b5
commit 991ad06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/onnx/features.py b/src/transformers/onnx/features.py
@@ -502,7 +502,7 @@ class FeaturesManager:
             onnx_config_cls="models.squeezebert.SqueezeBertOnnxConfig",
         ),
         "swin": supported_features_mapping(
-            "default", "image-classification", "masked-im", onnx_config_cls="models.swin.SwinOnnxConfig"
+            "default", "image-classification", onnx_config_cls="models.swin.SwinOnnxConfig"
         ),
         "t5": supported_features_mapping(
             "default",