mirror of
https://github.com/pstrueb/piper.git
synced 2026-05-29 16:15:51 +00:00
Merge pull request #172 from marty1885/fix-long-inference-gpu
Fix extremely long inference time when using CUDA with short sentences.
This commit is contained in:
@@ -41,7 +41,7 @@ class PiperVoice:
|
|||||||
sess_options=onnxruntime.SessionOptions(),
|
sess_options=onnxruntime.SessionOptions(),
|
||||||
providers=["CPUExecutionProvider"]
|
providers=["CPUExecutionProvider"]
|
||||||
if not use_cuda
|
if not use_cuda
|
||||||
else ["CUDAExecutionProvider"],
|
else [("CUDAExecutionProvider", {"cudnn_conv_algo_search": "HEURISTIC"})],
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user