Update config.py

Change the inference device for Mac to accelerate inference and reduce memory leak
2024-02-20 15:57:58 +00:00
parent a16de2e7c6
commit 3180294710
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"