SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix: data2vec-vision Onnx ready-made configuration. (#18427)

Browse Source 
* feat: add the data2vec conf that are missing https://huggingface.co/docs/transformers/serialization

* fix: wrong config
This commit is contained in:
Niklas Hansson
2022-08-09 09:35:05 +02:00
committed by GitHub
parent ab62a23d8c
commit fe785730dc
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/onnx/features.py
View File

@@ -229,6 +229,12 @@ class FeaturesManager:
"question-answering", "question-answering",
onnx_config_cls="models.data2vec.Data2VecTextOnnxConfig", onnx_config_cls="models.data2vec.Data2VecTextOnnxConfig",
), ),
"data2vec-vision": supported_features_mapping(
"default",
"image-classification",
"image-segmentation",
onnx_config_cls="models.data2vec.Data2VecVisionOnnxConfig",
),
"deberta": supported_features_mapping( "deberta": supported_features_mapping(
"default", "default",
"masked-lm", "masked-lm",

1
tests/onnx/test_onnx_v2.py
View File

@@ -207,6 +207,7 @@ PYTORCH_EXPORT_MODELS = {
("deit", "facebook/deit-small-patch16-224"), ("deit", "facebook/deit-small-patch16-224"),
("beit", "microsoft/beit-base-patch16-224"), ("beit", "microsoft/beit-base-patch16-224"),
("data2vec-text", "facebook/data2vec-text-base"), ("data2vec-text", "facebook/data2vec-text-base"),
("data2vec-vision", "facebook/data2vec-vision-base"),
("perceiver", "deepmind/language-perceiver", ("masked-lm", "sequence-classification")), ("perceiver", "deepmind/language-perceiver", ("masked-lm", "sequence-classification")),
("perceiver", "deepmind/vision-perceiver-conv", ("image-classification",)), ("perceiver", "deepmind/vision-perceiver-conv", ("image-classification",)),
("yolos", "hustvl/yolos-tiny"), ("yolos", "hustvl/yolos-tiny"),