From 64903abf772787947483530ac2c0926cdb74ffa0 Mon Sep 17 00:00:00 2001 From: jinzr Date: Tue, 5 Dec 2023 10:45:05 +0800 Subject: [PATCH] Update test_onnx.py --- egs/vctk/TTS/vits/test_onnx.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/egs/vctk/TTS/vits/test_onnx.py b/egs/vctk/TTS/vits/test_onnx.py index 8acca7c02..f76667e34 100755 --- a/egs/vctk/TTS/vits/test_onnx.py +++ b/egs/vctk/TTS/vits/test_onnx.py @@ -28,10 +28,11 @@ Use the onnx model to generate a wav: import argparse import logging +from pathlib import Path + import onnxruntime as ort import torch import torchaudio - from tokenizer import Tokenizer @@ -47,6 +48,12 @@ def get_parser(): help="Path to the onnx model.", ) + parser.add_argument( + "--speakers", + type=Path, + default=Path("data/speakers.txt"), + help="Path to speakers.txt file.", + ) parser.add_argument( "--tokens", type=str,