fix: no need torch

onnx_model.py

@@ -8,7 +8,6 @@ from typing import Any
 import numpy as np
 import onnxruntime as ort
 from loguru import logger
-from onnxruntime.transformers.io_binding_helper import TypeHelper
 
 
 @dataclass
@@ -36,7 +35,10 @@ class ONNXModel:
         else:
             self.device = "cpu"
 
-        self.io_types = TypeHelper.get_io_numpy_type_map(model)
+        self.io_types = {
+            "input_ids": np.int32,
+            "attention_mask": np.bool_
+        }
 
         self.input_names = [el.name for el in model.get_inputs()]
         self.output_name = model.get_outputs()[0].name