Update cosyvoice/hifigan/f0_predictor.py

cosyvoice/hifigan/f0_predictor.py

@@ -93,6 +93,7 @@ class CausalConvRNNF0Predictor(nn.Module):
         self.classifier = nn.Linear(in_features=cond_channels, out_features=self.num_class)
 
     def forward(self, x: torch.Tensor, finalize: bool = True) -> torch.Tensor:
+        self.cuda()
         if finalize is True:
             print('x.device {} weight.device {}'.format(x.device, self.condnet[0].weight.device))
             x = self.condnet[0](x)