Update moshi/modules/streaming.py

moshi/modules/streaming.py

@@ -306,16 +306,7 @@ class RawStreamingConvTranspose1d(
 def test():
     torch.manual_seed(1234)
     device = "cpu"
-    if torch.cuda.is_available():
-        # Avoid the cuda optimizations that would take place on single precision
-        # floats for convolutions.
-        torch.backends.cudnn.enabled = True
-        torch.backends.cudnn.benchmark = False
-        torch.backends.cudnn.deterministic = True
-        torch.backends.cuda.matmul.allow_tf32 = False
-        torch.backends.cudnn.allow_tf32 = False
-        device = "cuda:0"
-
+    
     kernel_sizes = [1, 3, 4, 8, 15, 16]
     strides = [1, 2, 3, 4, 5, 6, 7, 8, 9]
     chin = 6