298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-25 15:55:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Try to fix a vram issue with controlnets.

Browse Source
This commit is contained in:
comfyanonymous 2023-03-19 10:50:38 -04:00
parent 64ca71757a
commit edfc4ca663
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/sd.py
View File

@ -595,6 +595,9 @@ def load_controlnet(ckpt_path, model=None):
else: else:
control_model.load_state_dict(controlnet_data, strict=False) control_model.load_state_dict(controlnet_data, strict=False)
if use_fp16:
control_model = control_model.half()
control = ControlNet(control_model) control = ControlNet(control_model)
return control return control