diff --git a/egs/vctk/TTS/vits/export-onnx.py b/egs/vctk/TTS/vits/export-onnx.py index 2068adeea..94ec986f4 100755 --- a/egs/vctk/TTS/vits/export-onnx.py +++ b/egs/vctk/TTS/vits/export-onnx.py @@ -68,6 +68,12 @@ def get_parser(): help="The experiment dir", ) + parser.add_argument( + "--speakers", + type=Path, + default=Path("data/speakers.txt"), + help="Path to speakers.txt file.", + ) parser.add_argument( "--tokens", type=str, @@ -219,6 +225,10 @@ def main(): params.oov_id = tokenizer.oov_id params.vocab_size = tokenizer.vocab_size + with open(args.speakers) as f: + speaker_map = {line.strip(): i for i, line in enumerate(f)} + params.num_spks = len(speaker_map) + logging.info(params) logging.info("About to create model")