Add files via upload

This commit is contained in:
RVC-Boss
2023-10-08 18:31:09 +08:00
committed by GitHub
parent 361e674054
commit ec29cf8d65
3 changed files with 53 additions and 18 deletions

View File

@@ -9,7 +9,7 @@ import torch
from configs.config import Config
from infer.modules.uvr5.mdxnet import MDXNetDereverb
from infer.modules.uvr5.preprocess import AudioPre, AudioPreDeEcho
from infer.modules.uvr5.vr import AudioPre, AudioPreDeEcho
config = Config()
@@ -34,8 +34,9 @@ def uvr(model_name, inp_root, save_root_vocal, paths, save_root_ins, agg, format
os.getenv("weight_uvr5_root"), model_name + ".pth"
),
device=config.device,
is_half=config.is_half,
is_half=config.is_half
)
is_hp3 = "HP3" in model_name
if inp_root != "":
paths = [os.path.join(inp_root, name) for name in os.listdir(inp_root)]
else:
@@ -52,7 +53,7 @@ def uvr(model_name, inp_root, save_root_vocal, paths, save_root_ins, agg, format
):
need_reformat = 0
pre_fun._path_audio_(
inp_path, save_root_ins, save_root_vocal, format0
inp_path, save_root_ins, save_root_vocal, format0,is_hp3=is_hp3
)
done = 1
except: