debug threads

app.py

@@ -48,7 +48,8 @@ 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,6 +246,7 @@ class Vits_NT(Vits):
             y_mask=y_mask
         )
 
+        print('cuda available: ', torch.cuda.is_available())
         if not torch.cuda.is_available():
             num_chunks = 2
             chunk_size = z.shape[-1] // num_chunks