xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-12 11:46:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

save_checkpoint VAE optional

Browse Source 
Make VAE optional in save_checkpoint. Function argument has vae=None, but you call a method on the vae object without checking first if it is None.
This commit is contained in:
bash-j 2024-08-16 10:57:05 +09:30 committed by GitHub
parent bfc214f434
commit be9e8a64c9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/sd.py
View File

@ -668,7 +668,8 @@ def save_checkpoint(output_path, model, clip=None, vae=None, clip_vision=None, m
model_management.load_models_gpu(load_models, force_patch_weights=True)
clip_vision_sd = clip_vision.get_sd() if clip_vision is not None else None
sd = model.model.state_dict_for_saving(clip_sd, vae.get_sd(), clip_vision_sd)
vae = vae.get_sd() if vae is not None else None
sd = model.model.state_dict_for_saving(clip_sd, vae, clip_vision_sd)
for k in extra_keys:
sd[k] = extra_keys[k]