debug threads

app.py

@@ -48,8 +48,7 @@ hubert_model = HubertExtractor(
     device=device, 
     # storage_dir= # target storage dir hubert model
 )
-import os
-print("Available threads (CPU cores):", os.cpu_count())
+
 
 def get_manipulation(
     example,

pvq_manipulation/models/vits.py

@@ -246,9 +246,6 @@ class Vits_NT(Vits):
             y_mask=y_mask
         )
 
-        print('cuda available: ', torch.cuda.is_available())
-        import time
-        start = time.time()
         if not torch.cuda.is_available():
             num_chunks = min(os.cpu_count() or 2, z.shape[-1])
             chunk_size = z.shape[-1] // num_chunks
@@ -271,8 +268,6 @@ 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):