fix config path

infer_onnx.py

@@ -30,14 +30,14 @@ def process_text(i: int, text: str, device: torch.device):
 MODEL_PATH_MATCHA_MEL="matcha_multispeaker_cat_opset_15.onnx"
 MODEL_PATH_MATCHA="matcha_hifigan_multispeaker_cat.onnx"
 MODEL_PATH_VOCOS="mel_spec_22khz.onnx"
-CONFIG_PATH="/home/jgiraldo/projects/tts-onnx-comparison/config_22khz.yaml"
+CONFIG_PATH="config_22khz.yaml"
 
 sess_options = onnxruntime.SessionOptions()
 model_matcha_mel= onnxruntime.InferenceSession(str(MODEL_PATH_MATCHA_MEL), sess_options=sess_options, providers=["CPUExecutionProvider"])
 model_vocos = onnxruntime.InferenceSession(str(MODEL_PATH_VOCOS), sess_options=sess_options, providers=["CPUExecutionProvider"])
 model_matcha = onnxruntime.InferenceSession(str(MODEL_PATH_MATCHA), sess_options=sess_options, providers=["CPUExecutionProvider"])
 
-def vocos_inference(mel: torch.Tensor, config):
+def vocos_inference(mel: torch.Tensor):
 
     with open(CONFIG_PATH, "r") as f:
         config = yaml.safe_load(f)
@@ -102,7 +102,7 @@ def tts(text:str, spk_id:int):
 
     mel, mel_lengths = model_matcha_mel.run(None, inputs)
     # vocos inference
-    wavs_vocos = vocos_inference(mel, CONFIG_PATH)
+    wavs_vocos = vocos_inference(mel)
 
     with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp_matcha_vocos:
         sf.write(fp_matcha_vocos.name, wavs_vocos.squeeze(0), 22050, "PCM_24")