From 950fc4ce538f70980861e894c441d434163e3d91 Mon Sep 17 00:00:00 2001 From: Qing Date: Sat, 10 Feb 2024 16:24:40 +0800 Subject: [PATCH] fix controlnet torch_dtype --- iopaint/model/controlnet.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/iopaint/model/controlnet.py b/iopaint/model/controlnet.py index b908674..d52db01 100644 --- a/iopaint/model/controlnet.py +++ b/iopaint/model/controlnet.py @@ -90,6 +90,7 @@ class ControlNet(DiffusionInpaintModel): pretrained_model_name_or_path=controlnet_method, resume_download=True, local_files_only=model_kwargs["local_files_only"], + torch_dtype=self.torch_dtype, ) if model_info.is_single_file_diffusers: if self.model_info.model_type == ModelType.DIFFUSERS_SD: @@ -133,7 +134,10 @@ class ControlNet(DiffusionInpaintModel): def switch_controlnet_method(self, new_method: str): self.controlnet_method = new_method controlnet = ControlNetModel.from_pretrained( - new_method, resume_download=True, local_files_only=self.local_files_only + new_method, + resume_download=True, + local_files_only=self.local_files_only, + torch_dtype=self.torch_dtype, ).to(self.model.device) self.model.controlnet = controlnet