fix uvr5 hp3 output

This commit is contained in:
KamioRinn 2024-01-27 06:00:48 +08:00
parent 6571c1b3ba
commit f7130944c0
2 changed files with 5 additions and 2 deletions

View File

@ -110,6 +110,9 @@ class AudioPre:
y_spec_m = pred * X_phase
v_spec_m = X_spec_m - y_spec_m
if is_hp3 == True:
ins_root,vocal_root = vocal_root,ins_root
if ins_root is not None:
if self.data["high_end_process"].startswith("mirroring"):
input_high_end_ = spec_utils.mirroring(

View File

@ -81,7 +81,7 @@ def uvr(model_name, inp_root, save_root_vocal, paths, save_root_ins, agg, format
try:
if done == 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
)
infos.append("%s->Success" % (os.path.basename(inp_path)))
yield "\n".join(infos)
@ -89,7 +89,7 @@ def uvr(model_name, inp_root, save_root_vocal, paths, save_root_ins, agg, format
try:
if done == 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
)
infos.append("%s->Success" % (os.path.basename(inp_path)))
yield "\n".join(infos)