bacow/GPT-SoVITS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加健壮性

Browse Source
This commit is contained in:
chasonjiang
2024-03-13 20:08:32 +08:00
parent a116b55414
commit 03ae7fdb03
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
GPT_SoVITS/TTS_infer_pack/TTS.py
View File

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