Spaces:

fgnt-upb
/

pvq_manipulation

Running

FrederikRautenberg commited on 11 days ago

Commit

d794146

1 Parent(s): 2e528dd

debug threads

Files changed (1) hide show

pvq_manipulation/models/vits.py CHANGED Viewed

@@ -248,7 +248,7 @@ class Vits_NT(Vits):
         print('cuda available: ', torch.cuda.is_available())
         if not torch.cuda.is_available():
-            num_chunks = 2
             chunk_size = z.shape[-1] // num_chunks
             z_chunks = torch.split(z, chunk_size, dim=-1)

         print('cuda available: ', torch.cuda.is_available())
         if not torch.cuda.is_available():
+            num_chunks = 14
             chunk_size = z.shape[-1] // num_chunks
             z_chunks = torch.split(z, chunk_size, dim=-1)