aluminumbox commited on
Commit
f55db35
·
verified ·
1 Parent(s): 5faca47

Update cosyvoice/hifigan/generator.py

Browse files
Files changed (1) hide show
  1. cosyvoice/hifigan/generator.py +1 -1
cosyvoice/hifigan/generator.py CHANGED
@@ -365,7 +365,7 @@ class SourceModuleHnNSF(torch.nn.Module):
365
  # source for harmonic branch
366
  with torch.no_grad():
367
  sine_wavs, uv, _ = self.l_sin_gen(x)
368
- sine_merge = self.l_tanh(self.l_linear(sine_wavs))
369
 
370
  # source for noise branch, in the same shape as uv
371
  if self.training is False and self.causal is True:
 
365
  # source for harmonic branch
366
  with torch.no_grad():
367
  sine_wavs, uv, _ = self.l_sin_gen(x)
368
+ sine_merge = self.l_tanh(self.l_linear(sine_wavs.cuda()))
369
 
370
  # source for noise branch, in the same shape as uv
371
  if self.training is False and self.causal is True: