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
Hide 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:
onnx_config_cls="models.deberta_v2.DebertaV2OnnxConfig",
),
"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(
"default",
......@@ -515,7 +515,7 @@ class FeaturesManager:
"vision2seq-lm", onnx_config_cls="models.vision_encoder_decoder.VisionEncoderDecoderOnnxConfig"
),
"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(
"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