From 2cdb5844cb1fcda23e2fd2ec78c013f381e90e8d Mon Sep 17 00:00:00 2001 From: tsukumi Date: Mon, 23 Sep 2024 11:55:50 +0900 Subject: [PATCH] Fix: ONNX inference fails in Windows environments ref: https://qiita.com/gomasa/items/f6337087988685d00ebb --- style_bert_vits2/models/infer_onnx.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/style_bert_vits2/models/infer_onnx.py b/style_bert_vits2/models/infer_onnx.py index 725340a..e147395 100644 --- a/style_bert_vits2/models/infer_onnx.py +++ b/style_bert_vits2/models/infer_onnx.py @@ -92,9 +92,9 @@ def get_text_onnx( phone ), f"Bert seq len {bert.shape[-1]} != {len(phone)}" - phone = np.array(phone) - tone = np.array(tone) - language = np.array(language) + phone = np.array(phone, dtype=np.int64) + tone = np.array(tone, dtype=np.int64) + language = np.array(language, dtype=np.int64) return bert, ja_bert, en_bert, phone, tone, language