From 0f7511238524b2e752a6e04e1eb16402cc953941 Mon Sep 17 00:00:00 2001 From: zr_jin Date: Mon, 27 Jan 2025 16:10:44 +0800 Subject: [PATCH] Update dit.py --- egs/wenetspeech4tts/TTS/f5-tts/model/dit.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/egs/wenetspeech4tts/TTS/f5-tts/model/dit.py b/egs/wenetspeech4tts/TTS/f5-tts/model/dit.py index b048034707..966fabfdd4 100644 --- a/egs/wenetspeech4tts/TTS/f5-tts/model/dit.py +++ b/egs/wenetspeech4tts/TTS/f5-tts/model/dit.py @@ -92,9 +92,9 @@ def __init__(self, mel_dim, text_dim, out_dim): def forward( self, - x: float["b n d"], # noqa: F722 - cond: float["b n d"], # noqa: F722 - text_embed: float["b n d"], # noqa: F722 + x: float["b n d"], # noqa: F722 + cond: float["b n d"], # noqa: F722 + text_embed: float["b n d"], # noqa: F722 drop_audio_cond=False, ): if drop_audio_cond: # cfg for cond audio