From c140a0b691d9b1521b42fc5464c5e84a5b78d91c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Sat, 3 Aug 2024 17:11:08 +0800
Subject: [PATCH] feat(gpt): do re-gen only on None manual_seed

---
 ChatTTS/model/gpt.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ChatTTS/model/gpt.py b/ChatTTS/model/gpt.py
index 97661b0..89b0a7a 100644
--- a/ChatTTS/model/gpt.py
+++ b/ChatTTS/model/gpt.py
@@ -617,7 +617,7 @@ class GPT(nn.Module):
                     "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")