Skip to content

Commit 0624368

Browse files
Update optimum/exporters/openvino/model_configs.py
Co-authored-by: Ella Charlaix <[email protected]>
1 parent bd74200 commit 0624368

File tree

1 file changed

+1
-15
lines changed

1 file changed

+1
-15
lines changed

optimum/exporters/openvino/model_configs.py

Lines changed: 1 addition & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -4225,22 +4225,8 @@ class GPT2OpenVINOConfig(GPT2OnnxConfig):
42254225

42264226
@register_in_tasks_manager("dinov3_vit", *["feature-extraction"], library_name="transformers")
42274227
@register_in_tasks_manager("dinov3_convnext", *["feature-extraction"], library_name="transformers")
4228-
class DinoV3OpenVINOConfig(VisionOnnxConfig):
4228+
class DinoV3OpenVINOConfig(ViTOnnxConfig):
42294229
MIN_TRANSFORMERS_VERSION = "4.56.0"
4230-
NORMALIZED_CONFIG_CLASS = NormalizedVisionConfig
4231-
4232-
@property
4233-
def inputs(self) -> Dict[str, Dict[int, str]]:
4234-
return {
4235-
"pixel_values": {0: "image_batch_size", 1: "num_channels", 2: "height", 3: "width"},
4236-
}
4237-
4238-
@property
4239-
def outputs(self) -> Dict[str, Dict[int, str]]:
4240-
return {
4241-
"last_hidden_state": {0: "image_batch_size"},
4242-
"pooler_output": {0: "image_batch_size"},
4243-
}
42444230

42454231

42464232
@register_in_tasks_manager(

0 commit comments

Comments
 (0)