From d634c2727ec1723acccd47cc1cf4dd664a679382 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" <41898282+github-actions[bot]@users.noreply.github.com> Date: Fri, 1 Sep 2023 12:03:35 +0800 Subject: [PATCH] Format code (#1153) Co-authored-by: github-actions[bot] --- gui_v1.py | 39 +++++++++++++++++++++++++++++---------- tools/rvc_for_realtime.py | 12 ++++++------ 2 files changed, 35 insertions(+), 16 deletions(-) diff --git a/gui_v1.py b/gui_v1.py index 569d67f..97ddd08 100644 --- a/gui_v1.py +++ b/gui_v1.py @@ -303,8 +303,16 @@ if __name__ == "__main__": ), ], [ - sg.Checkbox(i18n("输入降噪"), key="I_noise_reduce", enable_events=True), - sg.Checkbox(i18n("输出降噪"), key="O_noise_reduce", enable_events=True), + sg.Checkbox( + i18n("输入降噪"), + key="I_noise_reduce", + enable_events=True, + ), + sg.Checkbox( + i18n("输出降噪"), + key="O_noise_reduce", + enable_events=True, + ), ], ], title=i18n("性能设置"), @@ -375,9 +383,9 @@ if __name__ == "__main__": json.dump(settings, j) if event == "stop_vc" and self.flag_vc == True: self.flag_vc = False - + # Parameter hot update - if event == 'threhold': + if event == "threhold": self.config.threhold = values["threhold"] elif event == "pitch": self.config.pitch = values["pitch"] @@ -389,14 +397,14 @@ if __name__ == "__main__": self.rvc.change_index_rate(values["index_rate"]) elif event in ["pm", "harvest", "crepe", "rmvpe"]: self.config.f0method = event - elif event == 'I_noise_reduce': + elif event == "I_noise_reduce": self.config.I_noise_reduce = values["I_noise_reduce"] - elif event == 'O_noise_reduce': + elif event == "O_noise_reduce": self.config.O_noise_reduce = values["O_noise_reduce"] elif event != "start_vc" and self.flag_vc == True: # Other parameters do not support hot update self.flag_vc = False - + def set_values(self, values): if len(values["pth_path"].strip()) == 0: sg.popup(i18n("请选择pth文件")) @@ -502,9 +510,20 @@ if __name__ == "__main__": self.sola_buffer: torch.Tensor = torch.zeros( self.crossfade_frame, device=device, dtype=torch.float32 ) - self.fade_in_window: torch.Tensor = torch.sin(0.5 * np.pi * torch.linspace( - 0.0, 1.0, steps=self.crossfade_frame, device=device, dtype=torch.float32 - )) ** 2 + self.fade_in_window: torch.Tensor = ( + torch.sin( + 0.5 + * np.pi + * torch.linspace( + 0.0, + 1.0, + steps=self.crossfade_frame, + device=device, + dtype=torch.float32, + ) + ) + ** 2 + ) self.fade_out_window: torch.Tensor = 1 - self.fade_in_window self.resampler = tat.Resample( orig_freq=self.config.samplerate, new_freq=16000, dtype=torch.float32 diff --git a/tools/rvc_for_realtime.py b/tools/rvc_for_realtime.py index a4a3a7b..f935c37 100644 --- a/tools/rvc_for_realtime.py +++ b/tools/rvc_for_realtime.py @@ -112,17 +112,17 @@ class RVC: self.is_half = config.is_half except: print(traceback.format_exc()) - + def change_key(self, new_key): self.f0_up_key = new_key - + def change_index_rate(self, new_index_rate): if new_index_rate != 0 and self.index_rate == 0: - self.index = faiss.read_index(self.index_path) - self.big_npy = self.index.reconstruct_n(0, self.index.ntotal) - print("index search enabled") + self.index = faiss.read_index(self.index_path) + self.big_npy = self.index.reconstruct_n(0, self.index.ntotal) + print("index search enabled") self.index_rate = new_index_rate - + def get_f0_post(self, f0): f0_min = self.f0_min f0_max = self.f0_max