Spaces:
Running on Zero
Running on Zero
Update cosyvoice/hifigan/generator.py
Browse files
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:
|