From eb686b8da3c6ff4228aef760d39e379995021d89 Mon Sep 17 00:00:00 2001 From: jinzr Date: Tue, 5 Dec 2023 14:38:35 +0800 Subject: [PATCH] Update test_onnx.py --- egs/vctk/TTS/vits/test_onnx.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/egs/vctk/TTS/vits/test_onnx.py b/egs/vctk/TTS/vits/test_onnx.py index f76667e34..4f8e9da19 100755 --- a/egs/vctk/TTS/vits/test_onnx.py +++ b/egs/vctk/TTS/vits/test_onnx.py @@ -111,6 +111,10 @@ def main(): tokenizer = Tokenizer(args.tokens) + with open(args.speakers) as f: + speaker_map = {line.strip(): i for i, line in enumerate(f)} + args.num_spks = len(speaker_map) + logging.info("About to create onnx model") model = OnnxModel(args.model_filename) @@ -118,7 +122,8 @@ def main(): tokens = tokenizer.texts_to_token_ids([text]) tokens = torch.tensor(tokens) # (1, T) tokens_lens = torch.tensor([tokens.shape[1]], dtype=torch.int64) # (1, T) - audio = model(tokens, tokens_lens) # (1, T') + speaker = torch.tensor([1], dtype=torch.int64) # (1, ) + audio = model(tokens, tokens_lens, speaker) # (1, T') torchaudio.save(str("test_onnx.wav"), audio, sample_rate=22050) logging.info("Saved to test_onnx.wav")