fix: index_root searching

close #1147
This commit is contained in:
源文雨
2023-09-01 14:11:55 +08:00
parent d634c2727e
commit 8ffdcb0128
21 changed files with 59 additions and 76 deletions

View File

@@ -101,7 +101,7 @@ class Predictor:
"CPUExecutionProvider",
],
)
print("onnx load done")
print("ONNX load done")
def demix(self, mix):
samples = mix.shape[-1]

View File

@@ -90,7 +90,7 @@ def uvr(model_name, inp_root, save_root_vocal, paths, save_root_ins, agg, format
del pre_fun
except:
traceback.print_exc()
print("clean_empty_cache")
if torch.cuda.is_available():
torch.cuda.empty_cache()
print("Executed torch.cuda.empty_cache()")
yield "\n".join(infos)