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

Update valle.py

Browse Source
This commit is contained in:
zr_jin 2024-12-07 18:19:29 +08:00
parent 6e6b022e41
commit e1e78e60f5
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/wenetspeech4tts/TTS/valle/valle.py
View File

@ -1686,8 +1686,8 @@ class VALLE(nn.Module):
decoder_outputs = decoder_outputs.to("cpu").type(torch.float32).detach().numpy() decoder_outputs = decoder_outputs.to("cpu").type(torch.float32).detach().numpy()
vmin, vmax = 0, 1024 # Encodec vmin, vmax = 0, 1024 # Encodec
if decoder_outputs.dtype == np.float32: # if decoder_outputs.dtype == np.float32:
vmin, vmax = -6, 0 # Fbank # vmin, vmax = -6, 0 # Fbank
num_figures = 3 num_figures = 3
for b, (utt_id, text) in enumerate(zip(utt_ids[:limit], texts[:limit])): for b, (utt_id, text) in enumerate(zip(utt_ids[:limit], texts[:limit])):