Update modeling_conformer.py

modeling_conformer.py

@@ -232,7 +232,8 @@ encoder.pre_encode.conv_module.{n},feature_extractor.conv_layers.{(n//3+1)}.conv
                     k = (b | ~v).view(1, -1, 1)
                     h = (torch.where(k, h[0], h2[0]), torch.where(k, h[1], h2[1]))
         self.cache_length = None
-        return [torch.tensor(i) for i in tok], [torch.tensor(i) for i in st]
+        device = next(self.parameters()).device
+        return [torch.tensor(i,device=device) for i in tok], [torch.tensor(i,device=device) for i in st]
 
     def make_srt(self, x, ts):
         t , s = x