test

Author: echarlaix

optimum/exporters/openvino/__main__.py

@@ -137,7 +137,6 @@ 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)
@@ -239,7 +238,7 @@ class StoreAttr(object):
         framework=framework,
         device=device,
         library_name=library_name,
-        torch_dtype=torch.float32,
+        # torch_dtype=torch.float32,
         **loading_kwargs,
     )