From 85d0d709e0a883dd89254929a5f510cc6bd0dfb2 Mon Sep 17 00:00:00 2001 From: Naozumi Date: Wed, 26 Jul 2023 11:54:37 +0800 Subject: [PATCH] Move `cor_nom`, `cor_den` to mps (`gui_v1.py`) (#851) * Move `cor_nom`, `cor_den` to mps * Split logic based on system --- gui_v1.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/gui_v1.py b/gui_v1.py index 5cfbc3b..0d45da6 100644 --- a/gui_v1.py +++ b/gui_v1.py @@ -556,9 +556,10 @@ if __name__ == "__main__": + 1e-8 ) if sys.platform == "darwin": - cor_nom = cor_nom.cpu() - cor_den = cor_den.cpu() - sola_offset = torch.argmax(cor_nom[0, 0] / cor_den[0, 0]) + _, sola_offset = torch.max(cor_nom[0, 0] / cor_den[0, 0]) + sola_offset = sola_offset.item() + else: + sola_offset = torch.argmax(cor_nom[0, 0] / cor_den[0, 0]) print("sola offset: " + str(int(sola_offset))) self.output_wav[:] = infer_wav[sola_offset : sola_offset + self.block_frame] self.output_wav[: self.crossfade_frame] *= self.fade_in_window