Skip to content

Commit fe78573

Browse files
authored
fix: data2vec-vision Onnx ready-made configuration. (#18427)
* feat: add the data2vec conf that are missing https://huggingface.co/docs/transformers/serialization * fix: wrong config
1 parent ab62a23 commit fe78573

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

src/transformers/onnx/features.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -229,6 +229,12 @@ class FeaturesManager:
229229
"question-answering",
230230
onnx_config_cls="models.data2vec.Data2VecTextOnnxConfig",
231231
),
232+
"data2vec-vision": supported_features_mapping(
233+
"default",
234+
"image-classification",
235+
"image-segmentation",
236+
onnx_config_cls="models.data2vec.Data2VecVisionOnnxConfig",
237+
),
232238
"deberta": supported_features_mapping(
233239
"default",
234240
"masked-lm",

tests/onnx/test_onnx_v2.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -207,6 +207,7 @@ def test_values_override(self):
207207
("deit", "facebook/deit-small-patch16-224"),
208208
("beit", "microsoft/beit-base-patch16-224"),
209209
("data2vec-text", "facebook/data2vec-text-base"),
210+
("data2vec-vision", "facebook/data2vec-vision-base"),
210211
("perceiver", "deepmind/language-perceiver", ("masked-lm", "sequence-classification")),
211212
("perceiver", "deepmind/vision-perceiver-conv", ("image-classification",)),
212213
("yolos", "hustvl/yolos-tiny"),

0 commit comments

Comments
 (0)