debug threads

pvq_manipulation/models/vits.py

@@ -247,8 +247,10 @@ class Vits_NT(Vits):
         )
 
         print('cuda available: ', torch.cuda.is_available())
+        import time
+        start = time.time()
         if not torch.cuda.is_available():
-            num_chunks = 14
+            num_chunks = min(os.cpu_count() or 2, z.shape[-1])
             chunk_size = z.shape[-1] // num_chunks
             z_chunks = torch.split(z, chunk_size, dim=-1)
 
@@ -269,6 +271,8 @@ class Vits_NT(Vits):
                 (z * y_mask)[:, :, : self.max_inference_len],
                 g=speaker_embedding_man[:, :, None] if self.config.gan_speaker_conditioning else None
             )
+        end = time.time()
+        print(end - start)
         return o
 
     def forward(self, x, x_lengths, y, y_lengths, aux_input, inference=False):