mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-10-07 07:14:52 +08:00
Compatible with default model name
This commit is contained in:
parent
104429d864
commit
1295bf22fb
@ -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:
|
||||
|
Loading…
x
Reference in New Issue
Block a user