feat: update config.py

config.py

@@ -1,4 +1,5 @@
 import argparse
+import sys
 import torch
 from multiprocessing import cpu_count
 
@@ -21,11 +22,10 @@ class Config:
 
     @staticmethod
     def arg_parse() -> tuple:
+        exe = sys.executable or "python"
         parser = argparse.ArgumentParser()
         parser.add_argument("--port", type=int, default=7865, help="Listen port")
-        parser.add_argument(
-            "--pycmd", type=str, default="python", help="Python command"
-        )
+        parser.add_argument("--pycmd", type=str, default=exe, help="Python command")
         parser.add_argument("--colab", action="store_true", help="Launch in colab")
         parser.add_argument(
             "--noparallel", action="store_true", help="Disable parallel processing"
@@ -49,6 +49,18 @@ class Config:
             cmd_opts.api
         )
 
+    # has_mps is only available in nightly pytorch (for now) and MasOS 12.3+.
+    # check `getattr` and try it for compatibility
+    @staticmethod
+    def has_mps() -> bool:
+        if not torch.backends.mps.is_available():
+            return False
+        try:
+            torch.zeros(1).to(torch.device("mps"))
+            return True
+        except Exception:
+            return False
+
     def device_config(self) -> tuple:
         if torch.cuda.is_available():
             i_device = int(self.device.split(":")[-1])
@@ -60,11 +72,10 @@ class Config:
                 or "1070" in self.gpu_name
                 or "1080" in self.gpu_name
             ):
-                print("16系/10系显卡和P40强制单精度")
+                print("Found GPU", self.gpu_name, ", force to fp32")
                 self.is_half = False
-
             else:
-                self.gpu_name = None
+                print("Found GPU", self.gpu_name)
             self.gpu_mem = int(
                 torch.cuda.get_device_properties(i_device).total_memory
                 / 1024
@@ -72,12 +83,12 @@ class Config:
                 / 1024
                 + 0.4
             )
-        elif torch.backends.mps.is_available():
-            print("没有发现支持的N卡, 使用MPS进行推理")
+        elif self.has_mps():
+            print("No supported Nvidia GPU found, use MPS instead")
             self.device = "mps"
             self.is_half = False
         else:
-            print("没有发现支持的N卡, 使用CPU进行推理")
+            print("No supported Nvidia GPU found, use CPU instead")
             self.device = "cpu"
             self.is_half = False