wetdog commited on
Commit
d3127d4
1 Parent(s): 35cd28c

fix input to vocos

Browse files

Mel spectrogram output from matcha is already a numpy ndarray rather than a torch tensor

Files changed (1) hide show
  1. 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: torch.Tensor):
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.float().numpy()
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