Skip to content

GitLab

Unverified Commit a72b82eb authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Remove a failing ONNX test (#23011) 



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 20ac86c6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/onnx/features.py
View file @ a72b82eb
...@@ -502,7 +502,7 @@ class FeaturesManager: ...@@ -502,7 +502,7 @@ class FeaturesManager:
onnx_config_cls="models.squeezebert.SqueezeBertOnnxConfig", onnx_config_cls="models.squeezebert.SqueezeBertOnnxConfig",
), ),
"swin": supported_features_mapping( "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( "t5": supported_features_mapping(
"default", "default",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment