Update cosyvoice/transformer/convolution.py

cosyvoice/transformer/convolution.py

@@ -177,6 +177,7 @@ 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)
         assert cache.size(2) == self.causal_padding
         if self.causal_type == 'left':
             x = torch.concat([cache, x], dim=2)