From 35c83e88dff3302534ea501ff7fec5588f52443b Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" <41898282+github-actions[bot]@users.noreply.github.com> Date: Thu, 25 Jan 2024 21:55:23 +0900 Subject: [PATCH] chore(format): run black on dev (#1763) Co-authored-by: github-actions[bot] --- gui_v1.py | 5 ++++- tools/rvc_for_realtime.py | 12 +++++++++--- 2 files changed, 13 insertions(+), 4 deletions(-) diff --git a/gui_v1.py b/gui_v1.py index 4331336..d5cd27c 100644 --- a/gui_v1.py +++ b/gui_v1.py @@ -530,7 +530,10 @@ if __name__ == "__main__": self.gui_config.sg_hostapi = self.hostapis[0] self.window["sg_hostapi"].Update(values=self.hostapis) self.window["sg_hostapi"].Update(value=self.gui_config.sg_hostapi) - if self.gui_config.sg_input_device not in self.input_devices and len(self.input_devices) > 0: + if ( + self.gui_config.sg_input_device not in self.input_devices + and len(self.input_devices) > 0 + ): self.gui_config.sg_input_device = self.input_devices[0] self.window["sg_input_device"].Update(values=self.input_devices) self.window["sg_input_device"].Update( diff --git a/tools/rvc_for_realtime.py b/tools/rvc_for_realtime.py index f0c6bb9..06e7056 100644 --- a/tools/rvc_for_realtime.py +++ b/tools/rvc_for_realtime.py @@ -375,14 +375,20 @@ class RVC: if (ix >= 0).all(): weight = np.square(1 / score) weight /= weight.sum(axis=1, keepdims=True) - npy = np.sum(self.big_npy[ix] * np.expand_dims(weight, axis=2), axis=1) + npy = np.sum( + self.big_npy[ix] * np.expand_dims(weight, axis=2), axis=1 + ) if self.config.is_half: npy = npy.astype("float16") feats[0][skip_head // 2 :] = ( - torch.from_numpy(npy).unsqueeze(0).to(self.device) * self.index_rate + torch.from_numpy(npy).unsqueeze(0).to(self.device) + * self.index_rate + (1 - self.index_rate) * feats[0][skip_head // 2 :] ) - else: printt("Invalid index. You MUST use added_xxxx.index but not trained_xxxx.index!") + else: + printt( + "Invalid index. You MUST use added_xxxx.index but not trained_xxxx.index!" + ) else: printt("Index search FAILED or disabled") except: