diff --git a/infer/modules/vc/pipeline.py b/infer/modules/vc/pipeline.py index 32d973b..b09ba94 100644 --- a/infer/modules/vc/pipeline.py +++ b/infer/modules/vc/pipeline.py @@ -264,9 +264,7 @@ class Pipeline(object): with torch.no_grad(): hasp = pitch is not None and pitchf is not None arg = (feats, p_len, pitch, pitchf, sid) if hasp else (feats, p_len, sid) - audio1 = ( - (net_g.infer(*arg)[0][0, 0]).data.cpu().float().numpy() - ) + audio1 = (net_g.infer(*arg)[0][0, 0]).data.cpu().float().numpy() del hasp, arg del feats, p_len, padding_mask if torch.cuda.is_available(): diff --git a/infer/modules/vc/utils.py b/infer/modules/vc/utils.py index 9517d32..c128707 100644 --- a/infer/modules/vc/utils.py +++ b/infer/modules/vc/utils.py @@ -2,6 +2,7 @@ import os from fairseq import checkpoint_utils + def get_index_path_from_model(sid): return next( ( diff --git a/tools/rvc_for_realtime.py b/tools/rvc_for_realtime.py index 350a128..5bd68ff 100644 --- a/tools/rvc_for_realtime.py +++ b/tools/rvc_for_realtime.py @@ -338,5 +338,14 @@ class RVC: .float() ) t5 = ttime() - print("Spent time: fea =", t2 - t1, ", index =", t3 - t2, ", f0 =", t4 - t3, ", model =", t5 - t4) + print( + "Spent time: fea =", + t2 - t1, + ", index =", + t3 - t2, + ", f0 =", + t4 - t3, + ", model =", + t5 - t4, + ) return infered_audio