Merge remote-tracking branch 'origin/4.0' into 4.0
# Conflicts: # webUI.py
This commit is contained in:
commit
efd3311da2
|
@ -6,6 +6,7 @@ import os
|
|||
import time
|
||||
from pathlib import Path
|
||||
from inference import slicer
|
||||
import gc
|
||||
|
||||
import librosa
|
||||
import numpy as np
|
||||
|
@ -221,6 +222,16 @@ class Svc(object):
|
|||
# 清理显存
|
||||
torch.cuda.empty_cache()
|
||||
|
||||
def unload_model(self):
|
||||
# 卸载模型
|
||||
self.net_g_ms = self.net_g_ms.to("cpu")
|
||||
del self.net_g_ms
|
||||
if hasattr(self,"enhancer"):
|
||||
self.enhancer.enhancer = self.enhancer.enhancer.to("cpu")
|
||||
del self.enhancer.enhancer
|
||||
del self.enhancer
|
||||
gc.collect()
|
||||
|
||||
def slice_inference(self,
|
||||
raw_audio_path,
|
||||
spk,
|
||||
|
|
24
webUI.py
24
webUI.py
|
@ -17,6 +17,7 @@ from scipy.io import wavfile
|
|||
import librosa
|
||||
import torch
|
||||
import time
|
||||
import traceback
|
||||
|
||||
logging.getLogger('numba').setLevel(logging.WARNING)
|
||||
logging.getLogger('markdown_it').setLevel(logging.WARNING)
|
||||
|
@ -28,15 +29,16 @@ model = None
|
|||
spk = None
|
||||
debug = False
|
||||
|
||||
cuda = []
|
||||
cuda = {}
|
||||
if torch.cuda.is_available():
|
||||
for i in range(torch.cuda.device_count()):
|
||||
device_name = torch.cuda.get_device_properties(i).name
|
||||
cuda.append(f"CUDA:{i} {device_name}")
|
||||
cuda[f"CUDA:{i} {device_name}"] = f"cuda:{i}"
|
||||
|
||||
def modelAnalysis(model_path,config_path,cluster_model_path,device,enhance):
|
||||
global model
|
||||
try:
|
||||
device = cuda[device] if "CUDA" in device else device
|
||||
model = Svc(model_path.name, config_path.name, device=device if device!="Auto" else None, cluster_model_path = cluster_model_path.name if cluster_model_path != None else "",nsf_hifigan_enhance=enhance)
|
||||
spks = list(model.spk2id.keys())
|
||||
device_name = torch.cuda.get_device_properties(model.dev).name if "cuda" in str(model.dev) else str(model.dev)
|
||||
|
@ -50,6 +52,7 @@ def modelAnalysis(model_path,config_path,cluster_model_path,device,enhance):
|
|||
msg += i + " "
|
||||
return sid.update(choices = spks,value=spks[0]), msg
|
||||
except Exception as e:
|
||||
if debug: traceback.print_exc()
|
||||
raise gr.Error(e)
|
||||
|
||||
|
||||
|
@ -58,6 +61,7 @@ def modelUnload():
|
|||
if model is None:
|
||||
return sid.update(choices = [],value=""),"没有模型需要卸载!"
|
||||
else:
|
||||
model.unload_model()
|
||||
model = None
|
||||
torch.cuda.empty_cache()
|
||||
return sid.update(choices = [],value=""),"模型卸载完毕!"
|
||||
|
@ -88,8 +92,10 @@ def vc_fn(sid, input_audio, vc_transform, auto_f0,cluster_ratio, slice_db, noise
|
|||
soundfile.write(output_file, _audio, model.target_sample, format="wav")
|
||||
return f"推理成功,音频文件保存为results/{filename}", (model.target_sample, _audio)
|
||||
except Exception as e:
|
||||
if debug: traceback.print_exc()
|
||||
raise gr.Error(e)
|
||||
except Exception as e:
|
||||
if debug: traceback.print_exc()
|
||||
raise gr.Error(e)
|
||||
|
||||
|
||||
|
@ -141,6 +147,9 @@ def vc_fn2(sid, input_audio, vc_transform, auto_f0,cluster_ratio, slice_db, nois
|
|||
os.remove(save_path2)
|
||||
return a,b
|
||||
|
||||
def debug_change():
|
||||
global debug
|
||||
debug = debug_button.value
|
||||
|
||||
with gr.Blocks(
|
||||
theme=gr.themes.Base(
|
||||
|
@ -162,7 +171,7 @@ with gr.Blocks(
|
|||
model_path = gr.File(label="选择模型文件")
|
||||
config_path = gr.File(label="选择配置文件")
|
||||
cluster_model_path = gr.File(label="选择聚类模型文件(没有可以不选)")
|
||||
device = gr.Dropdown(label="推理设备,默认为自动选择CPU和GPU", choices=["Auto",*cuda,"CPU"], value="Auto")
|
||||
device = gr.Dropdown(label="推理设备,默认为自动选择CPU和GPU", choices=["Auto",*cuda.keys(),"CPU"], value="Auto")
|
||||
enhance = gr.Checkbox(label="是否使用NSF_HIFIGAN增强,该选项对部分训练集少的模型有一定的音质增强效果,但是对训练好的模型有反面效果,默认关闭", value=False)
|
||||
with gr.Column():
|
||||
gr.Markdown(value="""
|
||||
|
@ -205,8 +214,15 @@ with gr.Blocks(
|
|||
vc_output1 = gr.Textbox(label="Output Message")
|
||||
with gr.Column():
|
||||
vc_output2 = gr.Audio(label="Output Audio", interactive=False)
|
||||
|
||||
with gr.Row(variant="panel"):
|
||||
with gr.Column():
|
||||
gr.Markdown(value="""
|
||||
<font size=2> WebUI设置</font>
|
||||
""")
|
||||
debug_button = gr.Checkbox(label="Debug模式,如果向社区反馈BUG需要打开,打开后控制台可以显示具体错误提示", value=debug)
|
||||
vc_submit.click(vc_fn, [sid, vc_input3, vc_transform,auto_f0,cluster_ratio, slice_db, noise_scale,pad_seconds,cl_num,lg_num,lgr_num,F0_mean_pooling,enhancer_adaptive_key], [vc_output1, vc_output2])
|
||||
vc_submit2.click(vc_fn2, [sid, vc_input3, vc_transform,auto_f0,cluster_ratio, slice_db, noise_scale,pad_seconds,cl_num,lg_num,lgr_num,text2tts,tts_rate,F0_mean_pooling,enhancer_adaptive_key], [vc_output1, vc_output2])
|
||||
debug_button.change(debug_change,[],[])
|
||||
vc_submit2.click(vc_fn2, [sid, vc_input3, vc_transform,auto_f0,cluster_ratio, slice_db, noise_scale,pad_seconds,cl_num,lg_num,lgr_num,text2tts,tts_rate,tts_voice,F0_mean_pooling,enhancer_adaptive_key], [vc_output1, vc_output2])
|
||||
model_load_button.click(modelAnalysis,[model_path,config_path,cluster_model_path,device,enhance],[sid,sid_output])
|
||||
model_unload_button.click(modelUnload,[],[sid,sid_output])
|
||||
|
|
Loading…
Reference in New Issue