tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Add: Support for speech synthesis in English and Chinese for ONNX inference in Non-JP-Extra models

Browse Source
This commit is contained in:
tsukumi
2024-09-23 07:14:44 +09:00
parent fe253611d3
commit dcca37b9a8
20 changed files with 342182 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
style_bert_vits2/nlp/japanese/bert_feature.py
View File

@@ -121,8 +121,8 @@ def extract_bert_feature_onnx(
res = session.run(
[output_name],
{
"input_ids": inputs["input_ids"].detach().numpy(),
"attention_mask": inputs["attention_mask"].detach().numpy(),
"input_ids": inputs["input_ids"].detach().numpy(), # type: ignore
"attention_mask": inputs["attention_mask"].detach().numpy(), # type: ignore
},
)[0]
@@ -132,8 +132,8 @@ def extract_bert_feature_onnx(
style_res = session.run(
[output_name],
{
"input_ids": style_inputs["input_ids"].detach().numpy(),
"attention_mask": style_inputs["attention_mask"].detach().numpy(),
"input_ids": style_inputs["input_ids"].detach().numpy(), # type: ignore
"attention_mask": style_inputs["attention_mask"].detach().numpy(), # type: ignore
},
)[0]
style_res_mean = np.mean(style_res, axis=0)