From 03ae7fdb03df551bc3ccaaf215e2ac598424edc0 Mon Sep 17 00:00:00 2001
From: chasonjiang <1440499136@qq.com>
Date: Wed, 13 Mar 2024 20:08:32 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=81=A5=E5=A3=AE=E6=80=A7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 GPT_SoVITS/TTS_infer_pack/TTS.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/GPT_SoVITS/TTS_infer_pack/TTS.py b/GPT_SoVITS/TTS_infer_pack/TTS.py
index e569f52..bcab7e8 100644
--- a/GPT_SoVITS/TTS_infer_pack/TTS.py
+++ b/GPT_SoVITS/TTS_infer_pack/TTS.py
@@ -672,6 +672,8 @@ class TTS:
                         "norm_text": norm_text,
                     }
                     batch_data.append(res)
+                if len(batch_data) == 0:
+                    return None
                 batch, _ = self.to_batch(batch_data, 
                             prompt_data=self.prompt_cache if not no_prompt_text else None, 
                             batch_size=batch_size, 
@@ -691,6 +693,8 @@ class TTS:
                 t3 = ttime()
                 if return_fragment:
                     item = make_batch(item)
+                    if item is None:
+                        continue
                     
                 batch_phones = item["phones"]
                 batch_phones_len = item["phones_len"]