youngkingdom/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-19 11:45:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Load metadata on VAELoader (#11846)

Browse Source 
Needed to load the proper LTX2 VAE if separated from checkpoint
This commit is contained in:
Jukka Seppänen
2026-01-14 00:37:21 +02:00
committed by GitHub
parent d9dc02a7d6
commit e4b4fb3479
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
nodes.py
View File

@ -798,8 +798,8 @@ class VAELoader:
vae_path = folder_paths.get_full_path_or_raise("vae_approx", vae_name)
else:
vae_path = folder_paths.get_full_path_or_raise("vae", vae_name)
sd = comfy.utils.load_torch_file(vae_path)
vae = comfy.sd.VAE(sd=sd)
sd, metadata = comfy.utils.load_torch_file(vae_path, return_metadata=True)
vae = comfy.sd.VAE(sd=sd, metadata=metadata)
vae.throw_exception_if_invalid()
return (vae,)