minor fixes

k2-fsa · Dec 6, 2024 · a8efe19 · a8efe19
1 parent 94126e7
commit a8efe19
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/egs/wenetspeech4tts/TTS/valle/valle.py b/egs/wenetspeech4tts/TTS/valle/valle.py
@@ -1676,7 +1676,8 @@ def visualize(
         text_tokens, text_tokens_lens = tokenizer(tokens)
         assert text_tokens.ndim == 2
 
-        utt_ids, texts = batch["utt_id"], batch["text"]
+        texts = batch["text"]
+        utt_ids = [cut.id for cut in batch["cut"]]
 
         encoder_outputs = predicts[0].to("cpu").type(torch.float32).detach().numpy()
         decoder_outputs = predicts[1]