From 94126e7f38c971ee02d8bd60dde3b5cc3ca743f6 Mon Sep 17 00:00:00 2001 From: zr_jin Date: Fri, 6 Dec 2024 13:57:59 +0800 Subject: [PATCH] Update valle.py --- egs/wenetspeech4tts/TTS/valle/valle.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/wenetspeech4tts/TTS/valle/valle.py b/egs/wenetspeech4tts/TTS/valle/valle.py index 206b843ba..0cf73926b 100644 --- a/egs/wenetspeech4tts/TTS/valle/valle.py +++ b/egs/wenetspeech4tts/TTS/valle/valle.py @@ -1669,8 +1669,8 @@ class VALLE(nn.Module): output_dir: str, limit: int = 4, ) -> None: - audio_features = batch["audio_features"].to("cpu").detach().numpy() - audio_features_lens = batch["audio_features_lens"].to("cpu").detach().numpy() + audio_features = batch["features"].to("cpu").detach().numpy() + audio_features_lens = batch["features_lens"].to("cpu").detach().numpy() tokens = batch["tokens"] text_tokens, text_tokens_lens = tokenizer(tokens)