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

make alternative fn

Browse Source
This commit is contained in:
ljleb 2024-08-06 23:02:05 -04:00
parent 7cd0cdfce6
commit 2e888691d6
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

@ -500,6 +500,9 @@ def load_checkpoint(config_path=None, ckpt_path=None, output_vae=True, output_cl
def load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, output_clipvision=False, embedding_directory=None, output_model=True):
sd = comfy.utils.load_torch_file(ckpt_path)
return load_state_dict_guess_config(sd, output_vae, output_clip, output_clipvision, embedding_directory, output_model)
def load_state_dict_guess_config(sd, output_vae=True, output_clip=True, output_clipvision=False, embedding_directory=None, output_model=True):
sd_keys = sd.keys()
clip = None
clipvision = None