Update config.py

精简代码
2024-02-21 01:15:31 +00:00
parent 220367f90c
commit db40317d9c
1 changed files with 0 additions and 2 deletions
--- a/config.py
+++ b/config.py
@@ -19,8 +19,6 @@ exp_root = "logs"
 python_exec = sys.executable or "python"
 if torch.cuda.is_available():
    infer_device = "cuda"
-elif torch.backends.mps.is_available():
-    infer_device = "cpu"
 else:
    infer_device = "cpu"