diff --git a/infer/lib/jit/synthesizer.py b/infer/lib/jit/synthesizer.py index 9a8bab1..6f40ffb 100644 --- a/infer/lib/jit/synthesizer.py +++ b/infer/lib/jit/synthesizer.py @@ -12,7 +12,9 @@ def get_synthesizer_ckpt(cpt, device=torch.device("cpu")): elif version == "v2": encoder_dim = 768 net_g = SynthesizerTrnMsNSFsid( - *cpt["config"], encoder_dim=encoder_dim, use_f0 = if_f0==1, + *cpt["config"], + encoder_dim=encoder_dim, + use_f0=if_f0 == 1, ) del net_g.enc_q net_g.load_state_dict(cpt["weight"], strict=False) diff --git a/rvc/onnx/infer.py b/rvc/onnx/infer.py index 52de675..31ac954 100644 --- a/rvc/onnx/infer.py +++ b/rvc/onnx/infer.py @@ -119,9 +119,7 @@ class RVC(Model): rnd = np.random.randn(1, 192, hubert_length).astype(np.float32) hubert_length = np.array([hubert_length]).astype(np.int64) - out_wav = self.forward( - hubert, hubert_length, pitch, pitchf, ds, rnd - ).squeeze() + out_wav = self.forward(hubert, hubert_length, pitch, pitchf, ds, rnd).squeeze() out_wav = np.pad(out_wav, (0, 2 * self.hop_len), "constant")