From 1295bf22fbc1e4dbc17a68bda6e16774c8cb44b3 Mon Sep 17 00:00:00 2001 From: KamioRinn Date: Mon, 15 Jul 2024 05:05:20 +0800 Subject: [PATCH] Compatible with default model name --- tools/uvr5/webui.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/uvr5/webui.py b/tools/uvr5/webui.py index df23522e..7749fbed 100644 --- a/tools/uvr5/webui.py +++ b/tools/uvr5/webui.py @@ -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 if model_name == "onnx_dereverb_By_FoxJoy": pre_fun = MDXNetDereverb(15) - elif model_name == "Bs_Roformer": + elif model_name == "Bs_Roformer" or "bs_roformer" in model_name.lower(): func = BsRoformer_Loader 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, ) else: