diff --git a/paddlespeech/t2s/frontend/g2pw/onnx_api.py b/paddlespeech/t2s/frontend/g2pw/onnx_api.py index 9628fa635d69b5f68c14c435687e56df3006b4d7..188d973721c5b740ecbcf35a722d055462c08f06 100644 --- a/paddlespeech/t2s/frontend/g2pw/onnx_api.py +++ b/paddlespeech/t2s/frontend/g2pw/onnx_api.py @@ -46,8 +46,10 @@ class G2PWOnnxConverter: uncompress_path = download_and_decompress(g2pw_onnx_models['G2PWModel']['1.0'],model_dir) sess_options = onnxruntime.SessionOptions() + sess_options.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_ALL + sess_options.execution_mode = onnxruntime.ExecutionMode.ORT_SEQUENTIAL sess_options.intra_op_num_threads = 2 - self.session_g2pW = onnxruntime.InferenceSession(os.path.join(model_dir, 'G2PWModel/g2pW.onnx')) + self.session_g2pW = onnxruntime.InferenceSession(os.path.join(model_dir, 'G2PWModel/g2pW.onnx'),sess_options=sess_options) self.config = load_config(os.path.join(model_dir, 'G2PWModel/config.py'), use_default=True) self.model_source = model_source if model_source else self.config.model_source