Update cosyvoice/hifigan/f0_predictor.py

cosyvoice/hifigan/f0_predictor.py

@@ -94,6 +94,7 @@ class CausalConvRNNF0Predictor(nn.Module):
 
     def forward(self, x: torch.Tensor, finalize: bool = True) -> torch.Tensor:
         if finalize is True:
+            print('{} {}'.format(x.device, self.condnet[0].weight.device))
             x = self.condnet[0](x)
         else:
             x = self.condnet[0](x[:, :, :-self.condnet[0].causal_padding], x[:, :, -self.condnet[0].causal_padding:])