mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-10 02:22:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update dit.py

Browse Source
This commit is contained in:
zr_jin 2025-01-27 16:10:44 +08:00 committed by GitHub
parent 525caf624c
commit 0f75112385
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/wenetspeech4tts/TTS/f5-tts/model/dit.py
View File

@ -92,9 +92,9 @@ class InputEmbedding(nn.Module):
def forward( def forward(
self, self,
x: float["b n d"], # noqa: F722 x: float["b n d"], # noqa: F722
cond: float["b n d"], # noqa: F722 cond: float["b n d"], # noqa: F722
text_embed: float["b n d"], # noqa: F722 text_embed: float["b n d"], # noqa: F722
drop_audio_cond=False, drop_audio_cond=False,
): ):
if drop_audio_cond: # cfg for cond audio if drop_audio_cond: # cfg for cond audio