From d397a588742297eeabc4373b0f5dd97b5d8583ed Mon Sep 17 00:00:00 2001 From: tsukumi Date: Sun, 6 Oct 2024 16:18:21 +0900 Subject: [PATCH] Fix: errors during ONNX CPU inference --- style_bert_vits2/models/infer_onnx.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/style_bert_vits2/models/infer_onnx.py b/style_bert_vits2/models/infer_onnx.py index 508c242..422b3de 100644 --- a/style_bert_vits2/models/infer_onnx.py +++ b/style_bert_vits2/models/infer_onnx.py @@ -204,9 +204,10 @@ def infer_onnx( ## 手元環境だと DmlExecutionProvider のみ常に空の辞書が返されるため、当面 onnx_providers から直接オプションを取り出している device_id = 0 onnx_providers_dict: dict[str, dict[str, Any]] = {} - for provider, options in onnx_providers: - if isinstance(options, dict): - onnx_providers_dict[provider] = options + for provider in onnx_providers: + if isinstance(provider, tuple): + provider_name, options = provider + onnx_providers_dict[provider_name] = options else: onnx_providers_dict[provider] = {} first_provider_options = onnx_providers_dict[first_provider]