blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-07 23:48:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

Eliminate punc risk

Browse Source
This commit is contained in:
KamioRinn 2024-03-30 15:22:08 +08:00
parent 914dedb3e1
commit dfd5f7dea8
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api.py
View File

@ -398,6 +398,10 @@ def cut_text(text, punc):
return text return text
def only_punc(text):
return not any(t.isalnum() or t.isalpha() for t in text)
def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language): def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language):
t0 = ttime() t0 = ttime()
prompt_text = prompt_text.strip("\n") prompt_text = prompt_text.strip("\n")
@ -425,6 +429,10 @@ def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language)
audio_bytes = BytesIO() audio_bytes = BytesIO()
for text in texts: for text in texts:
# 简单防止纯符号引发参考音频泄露
if only_punc(text):
continue
audio_opt = [] audio_opt = []
phones2, bert2, norm_text2 = get_phones_and_bert(text, text_language) phones2, bert2, norm_text2 = get_phones_and_bert(text, text_language)
bert = torch.cat([bert1, bert2], 1) bert = torch.cat([bert1, bert2], 1)