lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-13 08:59:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix controlnet

Browse Source 
#961
This commit is contained in:
layerdiffusion
2024-08-07 19:21:05 -07:00
parent 692a0b2422
commit 002341af5b
2 changed files with 33 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules_forge/supported_controlnet.py
View File

@@ -116,8 +116,8 @@ class ControlNetPatcher(ControlModelPatcher):
controlnet_config.pop("out_channels")
controlnet_config["hint_channels"] = controlnet_data["{}input_hint_block.0.weight".format(prefix)].shape[1]
with using_forge_operations():
control_model = cldm.ControlNet(**controlnet_config)
with using_forge_operations(dtype=unet_dtype):
control_model = cldm.ControlNet(**controlnet_config).to(dtype=unet_dtype)
if pth:
if 'difference' in controlnet_data: