Update dit.py

k2-fsa · Jan 27, 2025 · d679567 · d679567
1 parent 59cba78
commit d679567
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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)