From d6795678143d3e34523f10277a7c1fa82dff3910 Mon Sep 17 00:00:00 2001 From: zr_jin Date: Mon, 27 Jan 2025 15:57:05 +0800 Subject: [PATCH] Update dit.py --- egs/wenetspeech4tts/TTS/f5-tts/model/dit.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/egs/wenetspeech4tts/TTS/f5-tts/model/dit.py b/egs/wenetspeech4tts/TTS/f5-tts/model/dit.py index 9059757fe0..b048034707 100644 --- a/egs/wenetspeech4tts/TTS/f5-tts/model/dit.py +++ b/egs/wenetspeech4tts/TTS/f5-tts/model/dit.py @@ -92,11 +92,11 @@ def __init__(self, mel_dim, text_dim, out_dim): def forward( self, - x: float["b n d"], - cond: float["b n d"], - text_embed: float["b n d"], + 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, - ): # noqa: F722 + ): if drop_audio_cond: # cfg for cond audio cond = torch.zeros_like(cond)