test

Author: echarlaix

optimum/exporters/openvino/__main__.py

@@ -137,6 +137,7 @@ def main_export(
     >>> main_export("gpt2", output="gpt2_onnx/")
     ```
     """
+    device = "cpu"
     original_task = task
     task = TasksManager.map_from_synonym(task)
     framework = TasksManager.determine_framework(model_name_or_path, subfolder=subfolder, framework=framework)
@@ -237,6 +238,7 @@ class StoreAttr(object):
         framework=framework,
         device=device,
         library_name=library_name,
+        torch_dype=torch.float32,
         **loading_kwargs,
     )
 

optimum/exporters/openvino/convert.py

@@ -404,7 +404,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")
+            logger.warning(f"Export model to OpenVINO directly failed with: \n{ex}.\nModel will be exported to ONNX {dummy_inputs}")
 
             if stateful:
                 # cannot raise because stateful is enabled by default and it would break backward compatibility for models that couldn't convert to OV directly