From db40317d9ceaf782b5ccb383e044281a0489f29a Mon Sep 17 00:00:00 2001
From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com>
Date: Wed, 21 Feb 2024 01:15:31 +0000
Subject: [PATCH] Update config.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

精简代码
---
 config.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/config.py b/config.py
index caaadd4..1f74128 100644
--- 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"