diff --git a/egs/vctk/TTS/vits/train.py b/egs/vctk/TTS/vits/train.py index 59ecbf194..56f167a17 100755 --- a/egs/vctk/TTS/vits/train.py +++ b/egs/vctk/TTS/vits/train.py @@ -684,7 +684,7 @@ def compute_validation_loss( inner_model = model.module if isinstance(model, DDP) else model audio_pred, _, duration = inner_model.inference( text=tokens[0, : tokens_lens[0].item()], - sids=speakers, + sids=speakers[0], ) audio_pred = audio_pred.data.cpu().numpy() audio_len_pred = (