Update dit.py

k2-fsa · Jan 27, 2025 · 0f75112 · 0f75112
1 parent 525caf6
commit 0f75112
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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