blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2026-06-05 13:58:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: Added ReturnWay option to get_tts_wav

Browse Source
This commit is contained in:
Kaning123 2026-04-04 14:17:07 +08:00
parent 5c03499fcf
commit cb2b844f45
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
GPT_SoVITS/inference_webui.py
View File

@ -838,6 +838,8 @@ def get_tts_wav(
SaveOutputAsUndecoded=False, SaveOutputAsUndecoded=False,
SaveOutputAsUndecodedName="output.voice", SaveOutputAsUndecodedName="output.voice",
AddRandomSaltToSaveOutputAsUndecodedName=False, AddRandomSaltToSaveOutputAsUndecodedName=False,
ReturnWay = "yield", # "yield" or "return"
): ):
global cache global cache
if ref_wav_path: if ref_wav_path:
@ -1197,10 +1199,15 @@ def get_tts_wav(
audio_opt /= max_audio audio_opt /= max_audio
else: else:
audio_opt = audio_opt.cpu().detach().numpy() audio_opt = audio_opt.cpu().detach().numpy()
yield opt_sr, (audio_opt * 32767).astype(np.int16)
if ReturnWay == "yield":
yield opt_sr, (audio_opt * 32767).astype(np.int16)
else:
return opt_sr, (audio_opt * 32767).astype(np.int16)
def close_serv(): def close_serv():
if running_on == "local" if running_on == "local":
sys.exit(0) sys.exit(0)
else: else:
gr.Warning(i18n("服务器环境下该功能不可用")) gr.Warning(i18n("服务器环境下该功能不可用"))