Update models.py

2023-08-13 21:35:46 +08:00 · 2023-08-13 21:35:46 +08:00 · c67e9b63da
parent cbe54c34fc
commit c67e9b63da
1 changed files with 1144 additions and 1144 deletions
--- a/lib/infer_pack/models.py
+++ b/lib/infer_pack/models.py
@ -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