From e0f71939b9ba325c21262ac38f4a2b120f9b91c6 Mon Sep 17 00:00:00 2001 From: marty1885 Date: Wed, 9 Aug 2023 16:04:59 +0800 Subject: [PATCH] perf: use heuristic instead of exhaustiv conv algo search --- src/python_run/piper/voice.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/python_run/piper/voice.py b/src/python_run/piper/voice.py index 02a2ddd..1edda09 100644 --- a/src/python_run/piper/voice.py +++ b/src/python_run/piper/voice.py @@ -41,7 +41,7 @@ class PiperVoice: sess_options=onnxruntime.SessionOptions(), providers=["CPUExecutionProvider"] if not use_cuda - else ["CUDAExecutionProvider"], + else [("CUDAExecutionProvider", {"cudnn_conv_algo_search": "HEURISTIC"})], ), )