fix

2023-05-12 17:28:35 +08:00
parent ca470db570
commit 0363472adc
2 changed files with 2 additions and 2 deletions
--- 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):