blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-06 03:57:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: in case ignore too short text

Browse Source
This commit is contained in:
ling 2024-02-07 19:41:41 +08:00
parent 6469048de1
commit 2c1eb59283
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
GPT_SoVITS/inference_webui.py
View File

@ -342,6 +342,20 @@ def get_bert_final(phones, word2ph, text,language,device):
bert = torch.zeros((1024, len(phones))).to(device) bert = torch.zeros((1024, len(phones))).to(device)
return bert return bert
def merge_short_text_in_array(texts, threshold):
if (len(texts)) < 2:
return texts
result = []
text = ""
for ele in texts:
text += ele
if len(text) >= threshold:
result.append(text)
text = ""
if (len(text) > 0):
result[len(result)] += text
return result
def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language, how_to_cut=i18n("不切")): def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language, how_to_cut=i18n("不切")):
t0 = ttime() t0 = ttime()
prompt_language = dict_language[prompt_language] prompt_language = dict_language[prompt_language]
@ -394,6 +408,7 @@ def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language,
text = text.replace("\n\n", "\n") text = text.replace("\n\n", "\n")
print(i18n("实际输入的目标文本(切句后):"), text) print(i18n("实际输入的目标文本(切句后):"), text)
texts = text.split("\n") texts = text.split("\n")
texts = merge_short_text_in_array(texts, 5)
audio_opt = [] audio_opt = []
bert1=get_bert_final(phones1, word2ph1, norm_text1,prompt_language,device).to(dtype) bert1=get_bert_final(phones1, word2ph1, norm_text1,prompt_language,device).to(dtype)