fix input to vocos
Browse filesMel spectrogram output from matcha is already a numpy ndarray rather than a torch tensor
- infer_onnx.py +2 -2
infer_onnx.py
CHANGED
@@ -37,7 +37,7 @@ model_matcha_mel= onnxruntime.InferenceSession(str(MODEL_PATH_MATCHA_MEL), sess_
|
|
37 |
model_vocos = onnxruntime.InferenceSession(str(MODEL_PATH_VOCOS), sess_options=sess_options, providers=["CPUExecutionProvider"])
|
38 |
model_matcha = onnxruntime.InferenceSession(str(MODEL_PATH_MATCHA), sess_options=sess_options, providers=["CPUExecutionProvider"])
|
39 |
|
40 |
-
def vocos_inference(mel
|
41 |
|
42 |
with open(CONFIG_PATH, "r") as f:
|
43 |
config = yaml.safe_load(f)
|
@@ -52,7 +52,7 @@ def vocos_inference(mel: torch.Tensor):
|
|
52 |
mag, x, y = model_vocos.run(
|
53 |
None,
|
54 |
{
|
55 |
-
"mels": mel
|
56 |
},
|
57 |
)
|
58 |
|
|
|
37 |
model_vocos = onnxruntime.InferenceSession(str(MODEL_PATH_VOCOS), sess_options=sess_options, providers=["CPUExecutionProvider"])
|
38 |
model_matcha = onnxruntime.InferenceSession(str(MODEL_PATH_MATCHA), sess_options=sess_options, providers=["CPUExecutionProvider"])
|
39 |
|
40 |
+
def vocos_inference(mel):
|
41 |
|
42 |
with open(CONFIG_PATH, "r") as f:
|
43 |
config = yaml.safe_load(f)
|
|
|
52 |
mag, x, y = model_vocos.run(
|
53 |
None,
|
54 |
{
|
55 |
+
"mels": mel
|
56 |
},
|
57 |
)
|
58 |
|