From 0ef3da24c1ed03e1125802325355f5dacc8a25f9 Mon Sep 17 00:00:00 2001 From: jinzr Date: Thu, 30 Nov 2023 22:50:29 +0800 Subject: [PATCH] Update train.py --- egs/vctk/TTS/vits/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/vctk/TTS/vits/train.py b/egs/vctk/TTS/vits/train.py index 3fc637a46..d9f250cc9 100755 --- a/egs/vctk/TTS/vits/train.py +++ b/egs/vctk/TTS/vits/train.py @@ -338,7 +338,7 @@ def prepare_input( audio_lens = batch["audio_lens"].to(device) features_lens = batch["features_lens"].to(device) tokens = batch["tokens"] - speakers = torch.Tensor([speaker_map[sid] for sid in batch["speakers"]]).to(device) + speakers = torch.Tensor([speaker_map[sid] for sid in batch["speakers"]]).int().to(device) tokens = tokenizer.tokens_to_token_ids(tokens) tokens = k2.RaggedTensor(tokens)