Add onnx providers

This commit is contained in:
KamioRinn 2024-02-18 21:23:30 +08:00
parent c686192f26
commit 7def70d951

View File

@ -85,7 +85,7 @@ class G2PWOnnxConverter:
sess_options.intra_op_num_threads = 2
self.session_g2pW = onnxruntime.InferenceSession(
os.path.join(uncompress_path, 'g2pW.onnx'),
sess_options=sess_options)
sess_options=sess_options,providers=['CPUExecutionProvider'])
self.config = load_config(
config_path=os.path.join(uncompress_path, 'config.py'),
use_default=True)