Update valle.py

This commit is contained in:
zr_jin 2024-12-08 11:39:57 +08:00
parent 6d625b076b
commit d61f5a2833

View File

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