Improve: ONNX conversion logic improved, and support for noise_scale(_w) model argument

2024-09-22 16:14:25 +09:00
parent 9b27ba9777
commit 291e635e9e
3 changed files with 104 additions and 76 deletions
--- a/style_bert_vits2/models/infer_onnx.py
+++ b/style_bert_vits2/models/infer_onnx.py
@@ -167,6 +167,8 @@ def infer_onnx(
            style_vec_tensor,
            np.array([length_scale], dtype=np.float32),
            np.array([sdp_ratio], dtype=np.float32),
+            np.array([noise_scale], dtype=np.float32),
+            np.array([noise_scale_w], dtype=np.float32),
        ]

        first_provider = onnx_session.get_providers()[0]