Spaces:

FunAudioLLM
/

Fun-CosyVoice3-0.5B

Running on Zero

aluminumbox commited on 5 days ago

Commit

0707207

verified ·

1 Parent(s): 7315253

Update cosyvoice/hifigan/f0_predictor.py

Files changed (1) hide show

cosyvoice/hifigan/f0_predictor.py CHANGED Viewed

@@ -94,7 +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:])

     def forward(self, x: torch.Tensor, finalize: bool = True) -> torch.Tensor:
         if finalize is True:
+            print('x.device {} weight.device {}'.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:])