xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2025-12-30 18:04:00 +08: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:
control_model.load_state_dict(controlnet_data, strict=False)
if use_fp16:
control_model = control_model.half()
control = ControlNet(control_model)
return control