Move `cor_nom`, `cor_den` to mps (`gui_v1.py`) (#851)

* Move `cor_nom`, `cor_den` to mps * Split logic based on system
2023-07-26 11:54:37 +08:00 · 2023-07-26 11:54:37 +08:00 · 85d0d709e0
parent 78b8bfe890
commit 85d0d709e0
1 changed files with 4 additions and 3 deletions
--- 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