Skip to content

Commit 3702083

Browse files
committed
fix: wrong config
1 parent 7015ed1 commit 3702083

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/onnx/features.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,7 @@ class FeaturesManager:
233233
"default",
234234
"image-classification",
235235
"image-segmentation",
236-
onnx_config_cls="models.data2vec.Data2VecVisionConfig",
236+
onnx_config_cls="models.data2vec.Data2VecVisionOnnxConfig",
237237
),
238238
"deberta": supported_features_mapping(
239239
"default",

0 commit comments

Comments
 (0)