Skip to content

GitLab

Commit d8e58f0a authored by comfyanonymous's avatar comfyanonymous
Browse files

Detect hint_channels from controlnet.

parent 0cb14a33
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
comfy/sd.py
View file @ d8e58f0a
...@@ -878,7 +878,7 @@ def load_controlnet(ckpt_path, model=None): ...@@ -878,7 +878,7 @@ def load_controlnet(ckpt_path, model=None):
use_fp16 = model_management.should_use_fp16() use_fp16 = model_management.should_use_fp16()
controlnet_config = model_detection.model_config_from_unet(controlnet_data, prefix, use_fp16).unet_config controlnet_config = model_detection.model_config_from_unet(controlnet_data, prefix, use_fp16).unet_config
controlnet_config.pop("out_channels") controlnet_config.pop("out_channels")
controlnet_config["hint_channels"] = 3 controlnet_config["hint_channels"] = controlnet_data["{}input_hint_block.0.weight".format(prefix)].shape[1]
control_model = cldm.ControlNet(**controlnet_config) control_model = cldm.ControlNet(**controlnet_config)
if pth: if pth:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment