From 9bd0987815e432ff9cb8de359cfc8eca8f2312cf Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" <41898282+github-actions[bot]@users.noreply.github.com> Date: Mon, 14 Aug 2023 14:24:25 +0800 Subject: [PATCH] Format code (#1002) Co-authored-by: github-actions[bot] --- lib/infer_pack/models.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/lib/infer_pack/models.py b/lib/infer_pack/models.py index f94f5b1..4749738 100644 --- a/lib/infer_pack/models.py +++ b/lib/infer_pack/models.py @@ -414,19 +414,20 @@ class SourceModuleHnNSF(torch.nn.Module): self.l_tanh = torch.nn.Tanh() def forward(self, x, upp=None): - if hasattr(self,"ddtype")==False: - self.ddtype=self.l_linear.weight.dtype + if hasattr(self, "ddtype") == False: + self.ddtype = self.l_linear.weight.dtype sine_wavs, uv, _ = self.l_sin_gen(x, upp) # print(x.dtype,sine_wavs.dtype,self.l_linear.weight.dtype) # if self.is_half: # sine_wavs = sine_wavs.half() # sine_merge = self.l_tanh(self.l_linear(sine_wavs.to(x))) # print(sine_wavs.dtype,self.ddtype) - if(sine_wavs.dtype!=self.ddtype): - sine_wavs=sine_wavs.to(self.ddtype) + if sine_wavs.dtype != self.ddtype: + sine_wavs = sine_wavs.to(self.ddtype) sine_merge = self.l_tanh(self.l_linear(sine_wavs)) return sine_merge, None, None # noise, uv + class GeneratorNSF(torch.nn.Module): def __init__( self,