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

Update inference_webui.py

Browse Source
This commit is contained in:
RVC-Boss 2025-02-27 22:43:18 +08:00 committed by GitHub
parent a0ff6fa7a2
commit 94ffcbe616
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
GPT_SoVITS/inference_webui.py
View File

@ -690,7 +690,7 @@ def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language,
with torch.inference_mode(): with torch.inference_mode():
wav_gen = bigvgan_model(cmf_res) wav_gen = bigvgan_model(cmf_res)
audio=wav_gen[0][0]#.cpu().detach().numpy() audio=wav_gen[0][0]#.cpu().detach().numpy()
max_audio=torch.abs(audio).max()#简单防止16bit爆音#np.abs(audio).max() max_audio=torch.abs(audio).max()#简单防止16bit爆音
if max_audio>1:audio/=max_audio if max_audio>1:audio/=max_audio
audio_opt.append(audio) audio_opt.append(audio)
audio_opt.append(zero_wav_torch)#zero_wav audio_opt.append(zero_wav_torch)#zero_wav
@ -703,6 +703,8 @@ def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language,
if if_sr==True and sr==24000: if if_sr==True and sr==24000:
print(i18n("音频超分中")) print(i18n("音频超分中"))
audio_opt,sr=audio_sr(audio_opt.unsqueeze(0),sr) audio_opt,sr=audio_sr(audio_opt.unsqueeze(0),sr)
max_audio=np.abs(audio).max()
if max_audio > 1: audio /= max_audio
else: else:
audio_opt=audio_opt.cpu().detach().numpy() audio_opt=audio_opt.cpu().detach().numpy()
yield sr, (audio_opt * 32767).astype(np.int16) yield sr, (audio_opt * 32767).astype(np.int16)