Update config.py

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

View File

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