add run-sd-local arg

2022-09-29 13:13:09 +08:00
parent ec7b2d8e2d
commit 4d6cec0ca2
4 changed files with 22 additions and 13 deletions
--- a/lama_cleaner/model/sd.py
+++ b/lama_cleaner/model/sd.py
@@ -70,7 +70,7 @@ class SD(InpaintModel):
    def init_model(self, device: torch.device, **kwargs):
        from .sd_pipeline import StableDiffusionInpaintPipeline

-        model_kwargs = {}
+        model_kwargs = {"local_files_only": kwargs['sd_run_local']}
        if kwargs['sd_disable_nsfw']:
            logger.info("Disable Stable Diffusion Model NSFW checker")
            model_kwargs.update(dict(