"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "3520e37e86913715959ff14fef76340010c8de57"
Unverified Commit fe785730 authored by Niklas Hansson's avatar Niklas Hansson Committed by GitHub
Browse files

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
parent ab62a23d
...@@ -229,6 +229,12 @@ class FeaturesManager: ...@@ -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",
......
...@@ -207,6 +207,7 @@ PYTORCH_EXPORT_MODELS = { ...@@ -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"),
......
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