possible fix for when candidates >= samples

2023-10-10 15:30:08 +07:00 · 2023-10-10 15:30:08 +07:00 · 95f679f4ba
parent bf3b6c87aa
commit 95f679f4ba
1 changed files with 4 additions and 1 deletions
--- a/tortoise/api.py
+++ b/tortoise/api.py
@ -815,7 +815,10 @@ class TextToSpeech:

            clip_results = torch.cat(clip_results, dim=0)
            samples = torch.cat(samples, dim=0)
-            best_results = samples[torch.topk(clip_results, k=k).indices]
+            if k < num_autoregressive_samples:
+                best_results = samples[torch.topk(clip_results, k=k).indices]
+            else:
+                best_results = samples
            
            if not self.preloaded_tensors:
                self.clvp = migrate_to_device( self.clvp, 'cpu' )