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

fix

Browse Source
This commit is contained in:
Qing 2023-05-12 17:28:35 +08:00
parent ca470db570
commit 0363472adc
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lama_cleaner/model/sd.py
View File

@ -61,7 +61,7 @@ def load_from_local_model(local_model_path, torch_dtype, disable_nsfw=True):
del pipe del pipe
gc.collect() gc.collect()
return inpaint_pipe.to(torch_dtype) return inpaint_pipe.to(torch_dtype=torch_dtype)
class SD(DiffusionInpaintModel): class SD(DiffusionInpaintModel):

2
lama_cleaner/tests/test_model_md5.py
View File

@ -2,7 +2,7 @@ def test_load_model():
from lama_cleaner.plugins import InteractiveSeg from lama_cleaner.plugins import InteractiveSeg
from lama_cleaner.model_manager import ModelManager from lama_cleaner.model_manager import ModelManager
interactive_seg_model = InteractiveSeg() interactive_seg_model = InteractiveSeg('vit_l', 'cpu')
models = [ models = [
"lama", "lama",