From 49f0eb719762524cedfde7da440d32cf20d60ed0 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Fri, 29 Sep 2023 13:27:42 +0200 Subject: [PATCH] format --- optimum/exporters/openvino/convert.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/optimum/exporters/openvino/convert.py b/optimum/exporters/openvino/convert.py index 998fe228b4..916ac0df1a 100644 --- a/optimum/exporters/openvino/convert.py +++ b/optimum/exporters/openvino/convert.py @@ -307,9 +307,7 @@ def ts_patched_forward(*args, **kwargs): ov_model = convert_model(model, example_input=dummy_inputs, input=input_info) except Exception as ex: logger.warning(f"Export model to OpenVINO directly failed with: \n{ex}.\nModel will be exported to ONNX") - return export_pytorch_via_onnx( - model, config, opset, output, device, input_shapes, model_kwargs, fp16=fp16 - ) + return export_pytorch_via_onnx(model, config, opset, output, device, input_shapes, model_kwargs, fp16=fp16) ordered_dummy_inputs = {param: dummy_inputs[param] for param in sig.parameters if param in dummy_inputs} ordered_input_names = list(inputs) flatten_inputs = flattenize_inputs(ordered_dummy_inputs.values())