注释了inference_webui中mps的选项

2024-03-12 16:31:21 +08:00
parent 345f3203f8
commit f2cbc826c7
1 changed files with 2 additions and 2 deletions
--- a/GPT_SoVITS/inference_webui.py
+++ b/GPT_SoVITS/inference_webui.py
@@ -45,8 +45,8 @@ os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'  # 确保直接启动推理UI时

 if torch.cuda.is_available():
    device = "cuda"
-elif torch.backends.mps.is_available():
-    device = "mps"
+# elif torch.backends.mps.is_available():
+#     device = "mps"
 else:
    device = "cpu"