注释了inference_webui中mps的选项

This commit is contained in:
chasonjiang 2024-03-12 16:31:21 +08:00
parent 345f3203f8
commit f2cbc826c7

View File

@ -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"