fumiama/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-16 00:16:25 +08:00
Code Activity

Apply Code Formatter Change

Browse Source
This commit is contained in:
Tps-F
2023-08-19 10:58:39 +00:00
committed by github-actions[bot]
parent c25bb6c5d5
commit cb42c6990b
3 changed files with 18 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
infer/modules/uvr5/mdxnet.py
View File

@@ -13,7 +13,7 @@ cpu = torch.device("cpu")
class ConvTDFNetTrim:
def __init__(
self, device, model_name, target_name, L, dim_f, dim_t, n_fft, hop=1024
self, device, model_name, target_name, L, dim_f, dim_t, n_fft, hop=1024
):
super(ConvTDFNetTrim, self).__init__()
@@ -83,7 +83,7 @@ def get_models(device, dim_f, dim_t, n_fft):
dim_f=dim_f,
dim_t=dim_t,
n_fft=n_fft,
)
)
class Predictor:
@@ -95,7 +95,11 @@ class Predictor:
)
self.model = ort.InferenceSession(
os.path.join(args.onnx, self.model_.target_name + ".onnx"),
providers=["CUDAExecutionProvider", "DmlExecutionProvider", "CPUExecutionProvider"],
providers=[
"CUDAExecutionProvider",
"DmlExecutionProvider",
"CPUExecutionProvider",
],
)
print("onnx load done")
@@ -236,4 +240,4 @@ class MDXNetDereverb:
self.device = device
def path_audio(self, input, vocal_root, others_root, format):
self.pred.prediction(input, vocal_root, others_root, format)
self.pred.prediction(input, vocal_root, others_root, format)