diff --git a/iopaint/download.py b/iopaint/download.py index 7908d53..b55e5a6 100644 --- a/iopaint/download.py +++ b/iopaint/download.py @@ -36,10 +36,7 @@ def cli_download_model(model: str): from diffusers import DiffusionPipeline downloaded_path = handle_from_pretrained_exceptions( - DiffusionPipeline.download, - pretrained_model_name=model, - variant="fp16", - resume_download=True, + DiffusionPipeline.download, pretrained_model_name=model, variant="fp16" ) logger.info(f"Done. Downloaded to {downloaded_path}") diff --git a/iopaint/model/controlnet.py b/iopaint/model/controlnet.py index 7b4d243..24db2b6 100644 --- a/iopaint/model/controlnet.py +++ b/iopaint/model/controlnet.py @@ -79,6 +79,7 @@ class ControlNet(DiffusionInpaintModel): from diffusers import ( StableDiffusionControlNetInpaintPipeline as PipeClass, ) + original_config_file_name = "v1" elif model_info.model_type in [ @@ -88,11 +89,11 @@ class ControlNet(DiffusionInpaintModel): from diffusers import ( StableDiffusionXLControlNetInpaintPipeline as PipeClass, ) + original_config_file_name = "xl" controlnet = ControlNetModel.from_pretrained( pretrained_model_name_or_path=controlnet_method, - resume_download=True, local_files_only=model_kwargs["local_files_only"], torch_dtype=self.torch_dtype, ) @@ -139,7 +140,6 @@ class ControlNet(DiffusionInpaintModel): self.controlnet_method = new_method controlnet = ControlNetModel.from_pretrained( new_method, - resume_download=True, local_files_only=self.local_files_only, torch_dtype=self.torch_dtype, ).to(self.model.device)