Fix ONNX export

onnx_export.py

@@ -41,7 +41,11 @@ def convert_onnx(model_id: str, task: str, folder: str, opset: int) -> List:
     model_name = getattr(model, "name", None)
 
     onnx_config_constructor = TasksManager.get_exporter_config_constructor(
-        model_type, "onnx", task=task, model_name=model_name
+        exporter="onnx",
+        model=model,
+        task=task,
+        model_name=model_name,
+        model_type=model_type,
     )
     onnx_config = onnx_config_constructor(model.config)
 
@@ -66,12 +70,7 @@ def convert_onnx(model_id: str, task: str, folder: str, opset: int) -> List:
         opset = onnx_config.DEFAULT_ONNX_OPSET
 
     output = Path(folder).joinpath("model.onnx")
-    onnx_inputs, onnx_outputs = export(
-        model,
-        onnx_config,
-        opset,
-        output,
-    )
+    onnx_inputs, onnx_outputs = export(model, onnx_config, output, opset)
 
     atol = onnx_config.ATOL_FOR_VALIDATION
     if isinstance(atol, dict):