chore: change onnxruntime to 1.16

pipeline.py

@@ -20,7 +20,7 @@ class ONNXBaseModel(PreTrainedModel):
         if base_path:
             model_path = base_path + '/' + config.model_path
             if os.path.exists(model_path):
-                self.session = ort.InferenceSession(model_path)
+                self.session = ort.InferenceSession(model_path, providers=['CPUExecutionProvider'])
 
     def forward(self, input=None, **kwargs):
         outs = self.session.run(None, {'input': input})

requirements.txt

@@ -1,4 +1,4 @@
-transformers>=4.41.2
-torch>=2.3.1
-onnx>=1.16.1
-onnxruntime>=1.18.0
+transformers
+torch
+onnx
+onnxruntime==1.16.0