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

* Move `cor_nom`, `cor_den` to mps

* Split logic based on system

Author: Naozumi520

gui_v1.py

@@ -556,9 +556,10 @@ def audio_callback(
                 + 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