deepgeek/CosyVoice
2
0
Fork 0
mirror of https://github.com/FunAudioLLM/CosyVoice.git synced 2026-02-05 18:09:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix hifigan bug

Browse Source
This commit is contained in:
lyuxiang.lx
2025-02-18 14:45:43 +08:00
parent 890300513c
commit fcc054f64e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cosyvoice/hifigan/hifigan.py
View File

@@ -41,7 +41,7 @@ class HiFiGan(nn.Module):
loss_fm = feature_loss(fmap_rs, fmap_gs) loss_fm = feature_loss(fmap_rs, fmap_gs)
loss_mel = mel_loss(real_speech, generated_speech, self.mel_spec_transform) loss_mel = mel_loss(real_speech, generated_speech, self.mel_spec_transform)
if self.tpr_loss_weight != 0: if self.tpr_loss_weight != 0:
loss_tpr = tpr_loss(y_d_rs, y_d_gs, self.tpr_loss_tau) loss_tpr = tpr_loss(y_d_gs, y_d_rs, self.tpr_loss_tau)
else: else:
loss_tpr = torch.zeros(1).to(device) loss_tpr = torch.zeros(1).to(device)
loss_f0 = F.l1_loss(generated_f0, pitch_feat) loss_f0 = F.l1_loss(generated_f0, pitch_feat)