mirror of
https://github.com/pstrueb/piper.git
synced 2026-04-17 22:05:30 +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(),
|
||||
providers=["CPUExecutionProvider"]
|
||||
if not use_cuda
|
||||
else ["CUDAExecutionProvider"],
|
||||
else [("CUDAExecutionProvider", {"cudnn_conv_algo_search": "HEURISTIC"})],
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user