Fix: ONNX inference fails in Windows environments

ref: https://qiita.com/gomasa/items/f6337087988685d00ebb
This commit is contained in:
tsukumi
2024-09-23 11:55:50 +09:00
parent 53777e315e
commit 2cdb5844cb

View File

@@ -92,9 +92,9 @@ def get_text_onnx(
phone phone
), f"Bert seq len {bert.shape[-1]} != {len(phone)}" ), f"Bert seq len {bert.shape[-1]} != {len(phone)}"
phone = np.array(phone) phone = np.array(phone, dtype=np.int64)
tone = np.array(tone) tone = np.array(tone, dtype=np.int64)
language = np.array(language) language = np.array(language, dtype=np.int64)
return bert, ja_bert, en_bert, phone, tone, language return bert, ja_bert, en_bert, phone, tone, language