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

fix controlnet torch_dtype

Browse Source
This commit is contained in:
Qing 2024-02-10 16:24:40 +08:00
parent 170e7342f4
commit 950fc4ce53
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
iopaint/model/controlnet.py
View File

@ -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