Commit edfc4ca6 authored by comfyanonymous's avatar comfyanonymous
Browse files

Try to fix a vram issue with controlnets.

parent 64ca7175
......@@ -595,6 +595,9 @@ def load_controlnet(ckpt_path, model=None):
else:
control_model.load_state_dict(controlnet_data, strict=False)
if use_fp16:
control_model = control_model.half()
control = ControlNet(control_model)
return control
......
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