diff --git a/lama_cleaner/model/sd.py b/lama_cleaner/model/sd.py
index fec7fc6..945c506 100644
--- a/lama_cleaner/model/sd.py
+++ b/lama_cleaner/model/sd.py
@@ -61,7 +61,7 @@ def load_from_local_model(local_model_path, torch_dtype, disable_nsfw=True):
 
     del pipe
     gc.collect()
-    return inpaint_pipe.to(torch_dtype)
+    return inpaint_pipe.to(torch_dtype=torch_dtype)
 
 
 class SD(DiffusionInpaintModel):
diff --git a/lama_cleaner/tests/test_model_md5.py b/lama_cleaner/tests/test_model_md5.py
index 3bec003..1e50e46 100644
--- a/lama_cleaner/tests/test_model_md5.py
+++ b/lama_cleaner/tests/test_model_md5.py
@@ -2,7 +2,7 @@ def test_load_model():
     from lama_cleaner.plugins import InteractiveSeg
     from lama_cleaner.model_manager import ModelManager
 
-    interactive_seg_model = InteractiveSeg()
+    interactive_seg_model = InteractiveSeg('vit_l', 'cpu')
 
     models = [
         "lama",