feat(gpt): do re-gen only on None manual_seed

ericyue · Aug 3, 2024 · c140a0b · c140a0b
1 parent 61add70
commit c140a0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ChatTTS/model/gpt.py b/ChatTTS/model/gpt.py
@@ -617,7 +617,7 @@ def generate(
                     "unexpected end at index %s",
                     str([unexpected_idx.item() for unexpected_idx in finish.nonzero()]),
                 )
-                if ensure_non_empty:
+                if ensure_non_empty and manual_seed is None:
                     if show_tqdm:
                         pbar.close()
                     self.logger.warning("regenerate in order to ensure non-empty")