cleanup

Author: AznamirWoW

rvc/lib/algorithm/generators/refinegan.py

@@ -296,7 +296,6 @@ def __init__(
         upsample_initial_channel=512,
     ):
         super().__init__()
-        print('v2')
         self.upsample_rates = upsample_rates
         self.leaky_relu_slope = leaky_relu_slope
         self.checkpointing = checkpointing
@@ -388,7 +387,6 @@ def forward(self, mel: torch.Tensor, f0: torch.Tensor, g: torch.Tensor = None):
             self.upsample_conv_blocks,
             self.downsample_blocks,
         ):
-            # in-place call
             x = F.leaky_relu(x, self.leaky_relu_slope)
 
             if self.training and self.checkpointing:
@@ -400,10 +398,8 @@ def forward(self, mel: torch.Tensor, f0: torch.Tensor, g: torch.Tensor = None):
                 x = torch.cat([x, down(har_source)], dim=1)
                 x = res(x)
 
-        # in-place call
         x = F.leaky_relu(x, self.leaky_relu_slope)
         x = self.conv_post(x)
-        # in-place call
         x = torch.tanh(x)
 
         return x