Update inference_webui.py

添加程序入口,让推理方法可复用,修改DPO参数形参默认值与界面一致
This commit is contained in:
刘悦 2024-02-16 11:39:53 +08:00 committed by GitHub
parent 04c73c7016
commit 4afa9bf473
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -365,7 +365,7 @@ def merge_short_text_in_array(texts, threshold):
result[len(result) - 1] += text
return result
def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language, how_to_cut=i18n("不切"), top_k=20, top_p=0.6, temperature=0.6):
def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language, how_to_cut=i18n("不切"), top_k=5, top_p=1, temperature=1):
t0 = ttime()
prompt_language = dict_language[prompt_language]
text_language = dict_language[text_language]
@ -591,6 +591,8 @@ def get_weights_names():
SoVITS_names, GPT_names = get_weights_names()
def main():
with gr.Blocks(title="GPT-SoVITS WebUI") as app:
gr.Markdown(
value=i18n("本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责. <br>如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录<b>LICENSE</b>.")
@ -659,3 +661,6 @@ app.queue(concurrency_count=511, max_size=1022).launch(
server_port=infer_ttswebui,
quiet=True,
)
if __name__ == '__main__':
main()