From 991ad068c6406166c200978fec1f80b96a42c061 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 26 Apr 2023 17:44:12 +0200 Subject: [PATCH] Remove a failing ONNX test (#23011) fix Co-authored-by: ydshieh --- src/transformers/onnx/features.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/onnx/features.py b/src/transformers/onnx/features.py index b5ba7912c26569..1b0bf23d61213c 100644 --- 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",