fix: wrong config

huggingface · Aug 2, 2022 · 377c188 · 377c188
1 parent 8a8b76d
commit 377c188
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/onnx/features.py b/src/transformers/onnx/features.py
@@ -233,7 +233,7 @@ class FeaturesManager:
  "default",
  "image-classification",
  "image-segmentation",
- onnx_config_cls="models.data2vec.Data2VecVisionConfig",
+ onnx_config_cls="models.data2vec.Data2VecVisionOnnxConfig",
  ),
  "deberta": supported_features_mapping(
  "default",