This commit is contained in:
KamioRinn
2024-02-05 01:29:16 +08:00
parent e3d792fb56
commit 147218c529
2 changed files with 5 additions and 6 deletions

View File

@@ -239,7 +239,7 @@ class Predictor:
class MDXNetDereverb:
def __init__(self, chunks, device):
def __init__(self, chunks):
self.onnx = "%s/uvr5_weights/onnx_dereverb_By_FoxJoy"%os.path.dirname(os.path.abspath(__file__))
self.shifts = 10 # 'Predict with randomised equivariant stabilisation'
self.mixing = "min_mag" # ['default','min_mag','max_mag']
@@ -250,7 +250,7 @@ class MDXNetDereverb:
self.n_fft = 6144
self.denoise = True
self.pred = Predictor(self)
self.device = device
self.device = cpu
def _path_audio_(self, input, vocal_root, others_root, format, is_hp3=False):
self.pred.prediction(input, vocal_root, others_root, format)