Try to fix a vram issue with controlnets.

2023-03-19 10:50:38 -04:00 · 2023-03-19 10:50:38 -04:00 · edfc4ca663
parent 64ca71757a
commit edfc4ca663
1 changed files with 3 additions and 0 deletions
--- a/comfy/sd.py
+++ b/comfy/sd.py
@ -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