lj1995 commited on
Commit
167b457
·
verified ·
1 Parent(s): 79fea0b

Update text/g2pw/onnx_api.py

Browse files
Files changed (1) hide show
  1. text/g2pw/onnx_api.py +1 -4
text/g2pw/onnx_api.py CHANGED
@@ -90,10 +90,7 @@ class G2PWOnnxConverter:
90
  print(":::4")
91
  sess_options.intra_op_num_threads = 2
92
  print(":::5")
93
- try:
94
- self.session_g2pW = onnxruntime.InferenceSession(os.path.join(uncompress_path, 'g2pW.onnx'),sess_options=sess_options, providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
95
- except:
96
- self.session_g2pW = onnxruntime.InferenceSession(os.path.join(uncompress_path, 'g2pW.onnx'),sess_options=sess_options, providers=['CPUExecutionProvider'])
97
  print(":::6")
98
  self.config = load_config(
99
  config_path=os.path.join(uncompress_path, 'config.py'),
 
90
  print(":::4")
91
  sess_options.intra_op_num_threads = 2
92
  print(":::5")
93
+ self.session_g2pW = onnxruntime.InferenceSession(os.path.join(uncompress_path, 'g2pW.onnx'),sess_options=sess_options, providers=['CPUExecutionProvider'])
 
 
 
94
  print(":::6")
95
  self.config = load_config(
96
  config_path=os.path.join(uncompress_path, 'config.py'),