blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-25 02:58:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加引导音频记忆、引导音频文本记忆、语种记忆

Browse Source 
增加引导音频记忆、引导音频文本记忆、语种记忆
This commit is contained in:
刘悦 2024-01-26 15:40:14 +08:00 committed by GitHub
parent 4feb790895
commit ac35a77ebb
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
GPT_SoVITS/inference_webui.py
View File

@ -64,6 +64,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:
@ -183,6 +200,7 @@ dict_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")