Update config.py

Change the inference device for Mac to accelerate inference and reduce memory leak
2025-04-06 03:57:44 +08:00 · 2024-02-20 15:57:58 +00:00 · 2024-02-20 15:57:58 +00:00 · 3e81276f7c
commit 3e81276f7c
parent 31bd7c18a4
1 changed files with 1 additions and 1 deletions
--- a/config.py
+++ b/config.py
@ -20,7 +20,7 @@ python_exec = sys.executable or "python"
 if torch.cuda.is_available():
    infer_device = "cuda"
 elif torch.backends.mps.is_available():
-    infer_device = "mps"
+    infer_device = "cpu"
 else:
    infer_device = "cpu"