Update inference_webui.py

精简代码
This commit is contained in:
XXXXRT666 2024-02-21 01:15:11 +00:00 committed by GitHub
parent 861658050b
commit 220367f90c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -72,8 +72,6 @@ os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1' # 确保直接启动推理UI时
if torch.cuda.is_available():
device = "cuda"
elif torch.backends.mps.is_available():
device = "cpu"
else:
device = "cpu"