Load metadata on VAELoader (#11846)
Needed to load the proper LTX2 VAE if separated from checkpoint
This commit is contained in:
4
nodes.py
4
nodes.py
@@ -798,8 +798,8 @@ class VAELoader:
|
|||||||
vae_path = folder_paths.get_full_path_or_raise("vae_approx", vae_name)
|
vae_path = folder_paths.get_full_path_or_raise("vae_approx", vae_name)
|
||||||
else:
|
else:
|
||||||
vae_path = folder_paths.get_full_path_or_raise("vae", vae_name)
|
vae_path = folder_paths.get_full_path_or_raise("vae", vae_name)
|
||||||
sd = comfy.utils.load_torch_file(vae_path)
|
sd, metadata = comfy.utils.load_torch_file(vae_path, return_metadata=True)
|
||||||
vae = comfy.sd.VAE(sd=sd)
|
vae = comfy.sd.VAE(sd=sd, metadata=metadata)
|
||||||
vae.throw_exception_if_invalid()
|
vae.throw_exception_if_invalid()
|
||||||
return (vae,)
|
return (vae,)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user