Spaces:
Build error
Build error
Update text/g2pw/onnx_api.py
Browse files- 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 |
-
|
| 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'),
|