blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-05 19:41:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

在推理按钮下方增加推理使用参数

Browse Source
This commit is contained in:
Bobby 2024-03-19 09:36:29 +00:00
parent 5106fbc62a
commit a44d4a8b91
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
GPT_SoVITS/TTS_infer_pack/TTS.py
View File

@ -615,7 +615,7 @@ class TTS:
fragment_interval = inputs.get("fragment_interval", 0.3)
seed = inputs.get("seed", -1)
seed = -1 if seed in ["", None] else seed
set_seed(seed)
actual_seed = set_seed(seed)
if return_fragment:
# split_bucket = False
@ -824,13 +824,17 @@ class TTS:
if not return_fragment:
print("%.3f\t%.3f\t%.3f\t%.3f" % (t1 - t0, t2 - t1, t_34, t_45))
yield self.audio_postprocess(audio,
yield [
self.audio_postprocess(audio,
self.configs.sampling_rate,
batch_index_list,
speed_factor,
split_bucket,
fragment_interval
)
),
f"<strong>text:</strong> {text} <strong>text_lang:</strong> {text_lang} <strong>prompt_text:</strong> {prompt_text} <strong>prompt_lang:</strong> {prompt_lang} <strong>top_k:</strong> {top_k} <strong>top_p:</strong> {top_p} <strong>temperature:</strong> {temperature} <strong>batch_size:</strong> {batch_size} <strong>batch_threshold:</strong> {batch_threshold} <strong>split_bucket:</strong> {split_bucket} <strong>return_fragment:</strong> {return_fragment} <strong>speed_factor:</strong> {speed_factor} <strong>fragment_interval:</strong> {fragment_interval} <strong>seed:</strong> {actual_seed}"
]
except Exception as e:
traceback.print_exc()
# 必须返回一个空音频, 否则会导致显存不释放。

4
GPT_SoVITS/inference_webui.py
View File

@ -211,6 +211,8 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
with gr.Row():
inference_button = gr.Button(i18n("合成语音"), variant="primary")
stop_infer = gr.Button(i18n("终止合成"), variant="primary")
with gr.Row():
inference_details = gr.Markdown()
inference_button.click(
@ -224,7 +226,7 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
split_bucket,fragment_interval,
seed
],
[output],
[output, inference_details],
)
stop_infer.click(tts_pipline.stop, [], [])