Update models.py
This commit is contained in:
parent
cbe54c34fc
commit
c67e9b63da
|
@ -417,7 +417,7 @@ class SourceModuleHnNSF(torch.nn.Module):
|
||||||
sine_wavs, uv, _ = self.l_sin_gen(x, upp)
|
sine_wavs, uv, _ = self.l_sin_gen(x, upp)
|
||||||
if self.is_half:
|
if self.is_half:
|
||||||
sine_wavs = sine_wavs.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
|
return sine_merge, None, None # noise, uv
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue