nodes/lilly1987-ComfyUI_node_Lilly
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update VAELoaderText.py

Browse Source
This commit is contained in:
lilly
2023-10-25 00:34:18 +09:00
parent 70ed2501bb
commit 87d6fd0c35
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
VAELoaderText.py
+4 -1
View File
@@ -27,8 +27,11 @@ class VAELoaderText:
def load_vae(self, vae_name): def load_vae(self, vae_name):
print(f"[{ccolor}]vae_name : [/{ccolor}]", vae_name) print(f"[{ccolor}]vae_name : [/{ccolor}]", vae_name)
vae_path=getFullPath(vae_name,"vae") vae_path=getFullPath(vae_name,"vae")
try: try:
vae = comfy.sd.VAE(ckpt_path=vae_path) sd = comfy.utils.load_torch_file(vae_path)
vae = comfy.sd.VAE(sd=sd)
return (vae,) return (vae,)
except Exception as e: except Exception as e:
console.print_exception() console.print_exception()