From e4bcd6ac00c0721a5fec5154d0d9297241763884 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Wed, 1 Mar 2023 09:17:39 +0800
Subject: [PATCH] fix https://github.com/Sanster/lama-cleaner/issues/230

---
 lama_cleaner/helper.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/lama_cleaner/helper.py b/lama_cleaner/helper.py
index 3593010..ee44206 100644
--- a/lama_cleaner/helper.py
+++ b/lama_cleaner/helper.py
@@ -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