tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

修正多语言推理

Browse Source
This commit is contained in:
Stardust·减
2023-10-04 21:44:18 +08:00
committed by GitHub
parent 8e17a1fdd9
commit ae55007c5c
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
webui.py
View File

@@ -101,15 +101,16 @@ def infer(text, sdp_ratio, noise_scale, noise_scale_w, length_scale, sid, langua
.numpy()
)
del x_tst, tones, lang_ids, bert, x_tst_lengths, speakers
torch.cuda.empty_cache()
return audio
def tts_fn(text, speaker, sdp_ratio, noise_scale, noise_scale_w, length_scale):
def tts_fn(text, speaker, sdp_ratio, noise_scale, noise_scale_w, length_scale, language):
slices = text.split("|")
audio_list = []
with torch.no_grad():
for slice in slices:
audio = infer(slice, sdp_ratio=sdp_ratio, noise_scale=noise_scale, noise_scale_w=noise_scale_w, length_scale=length_scale, sid=speaker)
audio = infer(slice, sdp_ratio=sdp_ratio, noise_scale=noise_scale, noise_scale_w=noise_scale_w, length_scale=length_scale, sid=speaker, language=language)
audio_list.append(audio)
silence = np.zeros(hps.data.sampling_rate) # 生成1秒的静音
audio_list.append(silence) # 将静音添加到列表中