mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
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
This commit is contained in:
parent
ab62a23d8c
commit
fe785730dc
@ -229,6 +229,12 @@ class FeaturesManager:
|
||||
"question-answering",
|
||||
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(
|
||||
"default",
|
||||
"masked-lm",
|
||||
|
@ -207,6 +207,7 @@ PYTORCH_EXPORT_MODELS = {
|
||||
("deit", "facebook/deit-small-patch16-224"),
|
||||
("beit", "microsoft/beit-base-patch16-224"),
|
||||
("data2vec-text", "facebook/data2vec-text-base"),
|
||||
("data2vec-vision", "facebook/data2vec-vision-base"),
|
||||
("perceiver", "deepmind/language-perceiver", ("masked-lm", "sequence-classification")),
|
||||
("perceiver", "deepmind/vision-perceiver-conv", ("image-classification",)),
|
||||
("yolos", "hustvl/yolos-tiny"),
|
||||
|
Loading…
Reference in New Issue
Block a user