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

Merge pull request #216 from v3ucn/引导音频记忆，文本记忆和语种记忆

Browse Source 
引导音频记忆，文本记忆和语种记忆
This commit is contained in:
RVC-Boss 2024-01-26 17:50:16 +08:00 committed by GitHub
commit 35d4c334eb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
GPT_SoVITS/inference_webui.py
View File

@ -67,6 +67,23 @@ elif torch.backends.mps.is_available():
else: else:
device = "cpu" device = "cpu"
# 操作记忆功能
file_path = './audio_log.txt'
upload_audio_path = None
upload_audio_text = ""
upload_audio_lanuage = "中文"
if os.path.exists(file_path):
with open(file_path, 'r',encoding="utf-8") as file:
text_data = file.read()
text_data = text_data.split("|")
upload_audio_path = text_data[0]
upload_audio_text = text_data[1]
upload_audio_lanuage = text_data[2]
tokenizer = AutoTokenizer.from_pretrained(bert_path) tokenizer = AutoTokenizer.from_pretrained(bert_path)
bert_model = AutoModelForMaskedLM.from_pretrained(bert_path) bert_model = AutoModelForMaskedLM.from_pretrained(bert_path)
if is_half == True: if is_half == True:
@ -263,6 +280,7 @@ def nonen_get_bert_inf(text, language):
def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language): def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language):
with open("./audio_log.txt","w",encoding="utf-8")as f:f.write(f"{wav_path_log}|{prompt_text}|{prompt_language}")
t0 = ttime() t0 = ttime()
prompt_text = prompt_text.strip("\n") prompt_text = prompt_text.strip("\n")
prompt_language, text = prompt_language, text.strip("\n") prompt_language, text = prompt_language, text.strip("\n")
@ -482,10 +500,10 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
GPT_dropdown.change(change_gpt_weights,[GPT_dropdown],[]) GPT_dropdown.change(change_gpt_weights,[GPT_dropdown],[])
gr.Markdown(value=i18n("*请上传并填写参考信息")) gr.Markdown(value=i18n("*请上传并填写参考信息"))
with gr.Row(): with gr.Row():
inp_ref = gr.Audio(label=i18n("请上传参考音频"), type="filepath") inp_ref = gr.Audio(label=i18n("请上传参考音频"), type="filepath",value=upload_audio_path)
prompt_text = gr.Textbox(label=i18n("参考音频的文本"), value="") prompt_text = gr.Textbox(label=i18n("参考音频的文本"), value=upload_audio_text)
prompt_language = gr.Dropdown( prompt_language = gr.Dropdown(
label=i18n("参考音频的语种"),choices=[i18n("中文"),i18n("英文"),i18n("日文")],value=i18n("中文") label=i18n("参考音频的语种"),choices=[i18n("中文"),i18n("英文"),i18n("日文")],value=i18n(upload_audio_lanuage)
) )
gr.Markdown(value=i18n("*请填写需要合成的目标文本")) gr.Markdown(value=i18n("*请填写需要合成的目标文本"))
with gr.Row(): with gr.Row():