Skip to content

GitLab

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

Remove 2 failing ONNX conversion tests (#22660) 



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 06b05d45
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/transformers/onnx/features.py
View file @ ff73deeb
...@@ -267,7 +267,7 @@ class FeaturesManager: ...@@ -267,7 +267,7 @@ class FeaturesManager:
onnx_config_cls="models.deberta_v2.DebertaV2OnnxConfig", onnx_config_cls="models.deberta_v2.DebertaV2OnnxConfig",
), ),
"deit": supported_features_mapping( "deit": supported_features_mapping(
"default", "image-classification", "masked-im", onnx_config_cls="models.deit.DeiTOnnxConfig" "default", "image-classification", onnx_config_cls="models.deit.DeiTOnnxConfig"
), ),
"detr": supported_features_mapping( "detr": supported_features_mapping(
"default", "default",
...@@ -515,7 +515,7 @@ class FeaturesManager: ...@@ -515,7 +515,7 @@ class FeaturesManager:
"vision2seq-lm", onnx_config_cls="models.vision_encoder_decoder.VisionEncoderDecoderOnnxConfig" "vision2seq-lm", onnx_config_cls="models.vision_encoder_decoder.VisionEncoderDecoderOnnxConfig"
), ),
"vit": supported_features_mapping( "vit": supported_features_mapping(
"default", "image-classification", "masked-im", onnx_config_cls="models.vit.ViTOnnxConfig" "default", "image-classification", onnx_config_cls="models.vit.ViTOnnxConfig"
), ),
"whisper": supported_features_mapping( "whisper": 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