test

huggingface · Jul 15, 2024 · 22ce628 · 22ce628
1 parent ac9aa94
commit 22ce628
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/optimum_benchmark/backends/onnxruntime/backend.py b/optimum_benchmark/backends/onnxruntime/backend.py
@@ -292,11 +292,11 @@ def prepare_inputs(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
             with Accelerator().split_between_processes(inputs=inputs, apply_padding=False) as process_inputs:
                 inputs = process_inputs
 
-        if self.config.library == "transformers":
-            for key, value in list(inputs.items()):
-                if key in ["position_ids", "token_type_ids"]:
-                    if key not in self.pretrained_model.input_names:
-                        inputs.pop(key)
+        # if self.config.library == "transformers":
+        #     for key, value in list(inputs.items()):
+        #         if key in ["position_ids", "token_type_ids"]:
+        #             if key not in self.pretrained_model.input_names:
+        #                 inputs.pop(key)
 
         for key, value in inputs.items():
             if isinstance(value, torch.Tensor):