kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

fix https://github.com/Sanster/lama-cleaner/issues/230

Browse Source
This commit is contained in:
Qing 2023-03-01 09:17:39 +08:00
parent d1f36dde8e
commit e4bcd6ac00
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lama_cleaner/helper.py
View File

@ -23,9 +23,7 @@ def md5sum(filename):
def switch_mps_device(model_name, device):
if model_name not in MPS_SUPPORT_MODELS and (
device == "mps" or device == torch.device("mps")
):
if model_name not in MPS_SUPPORT_MODELS and str(device) == "mps":
logger.info(f"{model_name} not support mps, switch to cpu")
return torch.device("cpu")
return device