Update cosyvoice/transformer/convolution.py

cosyvoice/transformer/convolution.py

@@ -177,11 +177,12 @@ class CausalConv1d(torch.nn.Conv1d):
         input_timestep = x.shape[2]
         if cache.size(2) == 0:
             cache = torch.zeros(x.shape[0], x.shape[1], self.causal_padding).to(x)
-        cache = cache.to(x)
+        cache = cache.to(x.device)
         assert cache.size(2) == self.causal_padding
         if self.causal_type == 'left':
             x = torch.concat([cache, x], dim=2)
         else:
+            print('x.device {} cache.device {}'.format(x.device, cache.device))
             x = torch.concat([x, cache], dim=2)
         x = super(CausalConv1d, self).forward(x)
         assert x.shape[2] == input_timestep