diff --git a/infer/lib/rtrvc.py b/infer/lib/rtrvc.py index 8da568c..6e5a5e8 100644 --- a/infer/lib/rtrvc.py +++ b/infer/lib/rtrvc.py @@ -405,7 +405,10 @@ class RVC: if f0method == "rmvpe": f0_extractor_frame = 5120 * ((f0_extractor_frame - 1) // 5120 + 1) - 160 pitch, pitchf = self.get_f0( - input_wav[-f0_extractor_frame:], self.f0_up_key - self.formant_shift, self.n_cpu, f0method + input_wav[-f0_extractor_frame:], + self.f0_up_key - self.formant_shift, + self.n_cpu, + f0method, ) shift = block_frame_16k // 160 self.cache_pitch[:-shift] = self.cache_pitch[shift:].clone() @@ -413,7 +416,9 @@ class RVC: self.cache_pitch[4 - pitch.shape[0] :] = pitch[3:-1] self.cache_pitchf[4 - pitch.shape[0] :] = pitchf[3:-1] cache_pitch = self.cache_pitch[None, -p_len:] - cache_pitchf = self.cache_pitchf[None, -p_len:] * return_length2 / return_length + cache_pitchf = ( + self.cache_pitchf[None, -p_len:] * return_length2 / return_length + ) t4 = ttime() feats = F.interpolate(feats.permute(0, 2, 1), scale_factor=2).permute(0, 2, 1) feats = feats[:, :p_len, :]