From 0363472adc1c89fa33a6c4154d00da5c0269d615 Mon Sep 17 00:00:00 2001 From: Qing Date: Fri, 12 May 2023 17:28:35 +0800 Subject: [PATCH] fix --- lama_cleaner/model/sd.py | 2 +- lama_cleaner/tests/test_model_md5.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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",