Fix: ONNX inference fails in Windows environments
ref: https://qiita.com/gomasa/items/f6337087988685d00ebb
This commit is contained in:
@@ -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
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user