jaidaken/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

mp: Fix checkpoint saving (#12268)

Browse Source 
Fix regression in the recent model saving refactor. Pass the non unet
pieces down the layers so that checkpoints are complete.
This commit is contained in:
rattus
2026-02-03 23:08:45 -08:00
committed by GitHub
parent d30c609f5a
commit ef73070ea4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_patcher.py
View File

@@ -1400,7 +1400,7 @@ class ModelPatcher:
continue continue
key = "diffusion_model." + k key = "diffusion_model." + k
unet_state_dict[k] = LazyCastingParam(self, key, comfy.utils.get_attr(self.model, key)) unet_state_dict[k] = LazyCastingParam(self, key, comfy.utils.get_attr(self.model, key))
return self.model.state_dict_for_saving(unet_state_dict) return self.model.state_dict_for_saving(unet_state_dict, clip_state_dict=clip_state_dict, vae_state_dict=vae_state_dict, clip_vision_state_dict=clip_vision_state_dict)
def __del__(self): def __del__(self):
self.unpin_all_weights() self.unpin_all_weights()