blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-07 23:48:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

Compatible with default model name

Browse Source
This commit is contained in:
KamioRinn 2024-07-15 05:05:20 +08:00
parent 104429d864
commit 1295bf22fb
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/uvr5/webui.py
View File

@ -34,10 +34,10 @@ def uvr(model_name, inp_root, save_root_vocal, paths, save_root_ins, agg, format
is_hp3 = "HP3" in model_name is_hp3 = "HP3" in model_name
if model_name == "onnx_dereverb_By_FoxJoy": if model_name == "onnx_dereverb_By_FoxJoy":
pre_fun = MDXNetDereverb(15) pre_fun = MDXNetDereverb(15)
elif model_name == "Bs_Roformer": elif model_name == "Bs_Roformer" or "bs_roformer" in model_name.lower():
func = BsRoformer_Loader func = BsRoformer_Loader
pre_fun = func( pre_fun = func(
model_path = os.path.join(weight_uvr5_root, "Bs_Roformer.pth"), model_path = os.path.join(weight_uvr5_root, model_name + ".pth"),
device = device, device = device,
) )
else: else: