diff --git a/funasr/runtime/python/onnxruntime/rapid_paraformer/utils/utils.py b/funasr/runtime/python/onnxruntime/rapid_paraformer/utils/utils.py index 7943abbf9..392fe6b62 100644 --- a/funasr/runtime/python/onnxruntime/rapid_paraformer/utils/utils.py +++ b/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