Update models.py

This commit is contained in:
RVC-Boss
2023-08-13 21:35:46 +08:00
committed by GitHub
parent cbe54c34fc
commit c67e9b63da

View File

@@ -417,7 +417,7 @@ class SourceModuleHnNSF(torch.nn.Module):
sine_wavs, uv, _ = self.l_sin_gen(x, upp)
if self.is_half:
sine_wavs = sine_wavs.half()
sine_merge = self.l_tanh(self.l_linear(sine_wavs))
sine_merge = self.l_tanh(self.l_linear(sine_wavs.to(x)))
return sine_merge, None, None # noise, uv