rtf

funasr/runtime/python/onnxruntime/rapid_paraformer/utils/utils.py

@@ -150,6 +150,7 @@ class OrtInferSession():
     def __init__(self, model_file, device_id=-1):
         device_id = str(device_id)
         sess_opt = SessionOptions()
+        sess_opt.intra_op_num_threads = 4
         sess_opt.log_severity_level = 4
         sess_opt.enable_cpu_mem_arena = False
         sess_opt.graph_optimization_level = GraphOptimizationLevel.ORT_ENABLE_ALL