Get openvino exporter config

export.py

@@ -62,22 +62,18 @@ def convert_openvino(model_id: str, task: str, folder: str) -> List:
     if not isinstance(ov_model, OVStableDiffusionPipeline):
         try:
             model = TasksManager.get_model_from_task(task, model_id)
-            onnx_config_class = TasksManager.get_exporter_config_constructor(
-                exporter="onnx",
+            exporter_config_class = TasksManager.get_exporter_config_constructor(
+                exporter="openvino",
                 model=model,
                 task=task,
                 model_name=model_id,
                 model_type=model.config.model_type.replace("_", "-"),
             )
-            onnx_config = onnx_config_class(model.config)
-            inputs = onnx_config.generate_dummy_inputs(framework="pt")
+            openvino_config = exporter_config_class(model.config)
+            inputs = openvino_config.generate_dummy_inputs(framework="pt")
 
-            if isinstance(ov_model, (OVModelForCausalLM, OVModelForSeq2SeqLM)):
-                ov_outputs = ov_model.generate(**inputs)
-                outputs = model.generate(**inputs)
-            else:
-                ov_outputs = ov_model(**inputs)
-                outputs = model(**inputs)
+            ov_outputs = ov_model(**inputs)
+            outputs = model(**inputs)
 
             if isinstance(outputs, torch.Tensor):
                 outputs = {"logits": outputs}